Guanzheng commited on
Commit
4ba4097
1 Parent(s): 08636e2

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -30,7 +30,7 @@
30
  },
31
  "rope_theta": 10000,
32
  "tie_word_embeddings": false,
33
- "torch_dtype": "float32",
34
  "transformers_version": "4.29.1",
35
  "use_cache": true,
36
  "use_flashattn": true,
 
30
  },
31
  "rope_theta": 10000,
32
  "tie_word_embeddings": false,
33
+ "torch_dtype": "bfloat16",
34
  "transformers_version": "4.29.1",
35
  "use_cache": true,
36
  "use_flashattn": true,