jon-tow commited on
Commit
a2eb1af
1 Parent(s): ebf5f38

revert(config): use `float16` default torch dtype

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -17,7 +17,7 @@
17
  "partial_rotary_factor": 0.25,
18
  "rope_theta": 10000,
19
  "tie_word_embeddings": false,
20
- "torch_dtype": "bfloat16",
21
  "transformers_version": "4.38.0",
22
  "use_cache": true,
23
  "use_qkv_bias": true,
 
17
  "partial_rotary_factor": 0.25,
18
  "rope_theta": 10000,
19
  "tie_word_embeddings": false,
20
+ "torch_dtype": "float16",
21
  "transformers_version": "4.38.0",
22
  "use_cache": true,
23
  "use_qkv_bias": true,