Skip to content

Commit 2377ef5

Browse files
committed
repair deepseek_v2
1 parent 95f0253 commit 2377ef5

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)