01-ai
/

Yi-Coder-1.5B

Model card Files Files and versions Community

ayueei commited on Aug 27

Commit

9f832cb

•

1 Parent(s): 8d328a1

revise config

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -22,7 +22,7 @@
   "rope_scaling": null,
   "rope_theta": 10000000,
   "tie_word_embeddings": false,
-  "torch_dtype": "float32",
   "transformers_version": "4.44.0",
   "use_cache": true,
   "vocab_size": 64000

   "rope_scaling": null,
   "rope_theta": 10000000,
   "tie_word_embeddings": false,
+  "torch_dtype": "bfloat16",
   "transformers_version": "4.44.0",
   "use_cache": true,
   "vocab_size": 64000