jinliuxi
/

dsmini_pretrain

Text Generation

Model card Files Files and versions Community

jinliuxi commited on Mar 9

Commit

5e28bba

·

verified ·

1 Parent(s): cf352e7

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -50,7 +50,7 @@
   "tie_word_embeddings": false,
   "topk_group": 2,
   "topk_method": "noaux_tc",
-  "torch_dtype": "float32",
   "transformers_version": "4.48.3",
   "use_cache": true,
   "v_head_dim": 64,

   "tie_word_embeddings": false,
   "topk_group": 2,
   "topk_method": "noaux_tc",
+  "torch_dtype": "bfloat16",
   "transformers_version": "4.48.3",
   "use_cache": true,
   "v_head_dim": 64,