Update config.json
Browse files- config.json +1 -2
config.json
CHANGED
|
@@ -4,12 +4,11 @@
|
|
| 4 |
],
|
| 5 |
"auto_map": {
|
| 6 |
"AutoConfig": "configuration_ovis.OvisConfig",
|
| 7 |
-
"AutoModelForCausalLM": "
|
| 8 |
},
|
| 9 |
"conversation_formatter_class": "QwenConversationFormatter",
|
| 10 |
"disable_tie_weight": false,
|
| 11 |
"hidden_size": 1536,
|
| 12 |
-
"llm_attn_implementation": "flash_attention_2",
|
| 13 |
"llm_config": {
|
| 14 |
"_attn_implementation_autoset": true,
|
| 15 |
"_name_or_path": "Qwen/Qwen2.5-1.5B-Instruct",
|
|
|
|
| 4 |
],
|
| 5 |
"auto_map": {
|
| 6 |
"AutoConfig": "configuration_ovis.OvisConfig",
|
| 7 |
+
"AutoModelForCausalLM": "modeling_ovis.Ovis"
|
| 8 |
},
|
| 9 |
"conversation_formatter_class": "QwenConversationFormatter",
|
| 10 |
"disable_tie_weight": false,
|
| 11 |
"hidden_size": 1536,
|
|
|
|
| 12 |
"llm_config": {
|
| 13 |
"_attn_implementation_autoset": true,
|
| 14 |
"_name_or_path": "Qwen/Qwen2.5-1.5B-Instruct",
|