Skip to content

Commit e56d096

Browse files
authored
重做deepseek_v2模型并补注册 (#1859)
1 parent 95f0253 commit e56d096

File tree

2 files changed

+291
-149
lines changed

2 files changed

+291
-149
lines changed

mindnlp/transformers/models/auto/modeling_auto.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -468,6 +468,7 @@
468468
("ctrl", "CTRLLMHeadModel"),
469469
("data2vec-text", "Data2VecTextForCausalLM"),
470470
("dbrx", "DbrxForCausalLM"),
471+
("deepseek_v2", "DeepseekV2ForCausalLM"),
471472
("electra", "ElectraForCausalLM"),
472473
("ernie", "ErnieForCausalLM"),
473474
("falcon", "FalconForCausalLM"),

0 commit comments

Comments
 (0)