Org-Exp
/

SSM-350m

Parveshiiii commited on 8 days ago

Commit

44cd296

verified ·

1 Parent(s): 7667a23

Update config.json

Files changed (1) hide show

config.json CHANGED Viewed

@@ -75,7 +75,7 @@
   "router_aux_loss_coef": 0.01,
   "shared_intermediate_size": 2048,
   "tie_word_embeddings": true,
-  "torch_dtype": "bfloat16",
   "transformers_version": "4.56.0",
   "use_cache": true,
   "vocab_size": 100352

   "router_aux_loss_coef": 0.01,
   "shared_intermediate_size": 2048,
   "tie_word_embeddings": true,
+  "torch_dtype": "float32",
   "transformers_version": "4.56.0",
   "use_cache": true,
   "vocab_size": 100352