adalbertojunior commited on
Commit
9696660
·
verified ·
1 Parent(s): 5c745cf

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -3,7 +3,7 @@
3
  "Qwen2ForMaskedLM"
4
  ],
5
  "auto_map": {
6
- "AutoModel": "modeling_qwen.DusMistralModel",
7
  "AutoModelForCausalLM": "modeling_qwen.Qwen2ForCausalLM",
8
  "AutoModelForMaskedLM": "modeling_qwen.Qwen2ForMaskedlLM",
9
  "AutoModelForTokenClassification": "modeling_qwen.Qwen2ForTokenClassification",
 
3
  "Qwen2ForMaskedLM"
4
  ],
5
  "auto_map": {
6
+ "AutoModel": "modeling_qwen.Qwen2Model",
7
  "AutoModelForCausalLM": "modeling_qwen.Qwen2ForCausalLM",
8
  "AutoModelForMaskedLM": "modeling_qwen.Qwen2ForMaskedlLM",
9
  "AutoModelForTokenClassification": "modeling_qwen.Qwen2ForTokenClassification",