Fix config.json
Browse files- config.json +3 -3
config.json
CHANGED
|
@@ -8,9 +8,9 @@
|
|
| 8 |
"AutoConfig": "configuration_qwen.QWenConfig",
|
| 9 |
"AutoModelForCausalLM": "modeling_qwen.QWenLMHeadModel"
|
| 10 |
},
|
| 11 |
-
"bf16":
|
| 12 |
"emb_dropout_prob": 0.0,
|
| 13 |
-
"fp16":
|
| 14 |
"fp32": false,
|
| 15 |
"hidden_size": 4096,
|
| 16 |
"initializer_range": 0.02,
|
|
@@ -64,7 +64,7 @@
|
|
| 64 |
"use_cache_kernel": false,
|
| 65 |
"use_cache_quantization": false,
|
| 66 |
"use_dynamic_ntk": true,
|
| 67 |
-
"use_flash_attn":
|
| 68 |
"use_logn_attn": true,
|
| 69 |
"vocab_size": 151936
|
| 70 |
}
|
|
|
|
| 8 |
"AutoConfig": "configuration_qwen.QWenConfig",
|
| 9 |
"AutoModelForCausalLM": "modeling_qwen.QWenLMHeadModel"
|
| 10 |
},
|
| 11 |
+
"bf16": false,
|
| 12 |
"emb_dropout_prob": 0.0,
|
| 13 |
+
"fp16": true,
|
| 14 |
"fp32": false,
|
| 15 |
"hidden_size": 4096,
|
| 16 |
"initializer_range": 0.02,
|
|
|
|
| 64 |
"use_cache_kernel": false,
|
| 65 |
"use_cache_quantization": false,
|
| 66 |
"use_dynamic_ntk": true,
|
| 67 |
+
"use_flash_attn": "auto",
|
| 68 |
"use_logn_attn": true,
|
| 69 |
"vocab_size": 151936
|
| 70 |
}
|