zeio commited on
Commit
03d5b37
·
verified ·
1 Parent(s): 00532e9

fix(dtype): changed dtype from bfloat16 to float16

Browse files
Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -21,7 +21,7 @@
21
  "rope_theta": 10000.0,
22
  "sliding_window": 4096,
23
  "tie_word_embeddings": false,
24
- "torch_dtype": "bfloat16",
25
  "transformers_version": "4.37.0.dev0",
26
  "use_cache": true,
27
  "vocab_size": 32002,
@@ -54,4 +54,4 @@
54
  ]
55
  ]
56
  }
57
- }
 
21
  "rope_theta": 10000.0,
22
  "sliding_window": 4096,
23
  "tie_word_embeddings": false,
24
+ "torch_dtype": "float16",
25
  "transformers_version": "4.37.0.dev0",
26
  "use_cache": true,
27
  "vocab_size": 32002,
 
54
  ]
55
  ]
56
  }
57
+ }