tyj2022 mkshing commited on
Commit
78cad5a
1 Parent(s): ceadefe

change dtype to bf16 in config.json (#1)

Browse files

- change dtype to bf16 in config.json (b5f2f039e04a22eaf0273760c4b9fa62faaa2635)


Co-authored-by: Makoto Shing <mkshing@users.noreply.huggingface.co>

Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -24,7 +24,7 @@
24
  "rope_theta": 10000.0,
25
  "sliding_window": 4096,
26
  "tie_word_embeddings": false,
27
- "torch_dtype": "float32",
28
  "transformers_version": "4.38.2",
29
  "use_cache": false,
30
  "vocab_size": 32000
 
24
  "rope_theta": 10000.0,
25
  "sliding_window": 4096,
26
  "tie_word_embeddings": false,
27
+ "torch_dtype": "bfloat16",
28
  "transformers_version": "4.38.2",
29
  "use_cache": false,
30
  "vocab_size": 32000