seba
/

Dia-1.6B-bfloat16

model_hub_mixin

pytorch_model_hub_mixin

Model card Files Files and versions Community

seba commited on May 15

Commit

4478e21

·

verified ·

1 Parent(s): 0b0a127

set weight_dtype to bfloat16

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -43,7 +43,7 @@
     "rope_min_timescale": 1,
     "src_vocab_size": 256,
     "tgt_vocab_size": 1028,
-    "weight_dtype": "float32"
   },
   "training": {},
   "version": "0.1"

     "rope_min_timescale": 1,
     "src_vocab_size": 256,
     "tgt_vocab_size": 1028,
+    "weight_dtype": "bfloat16"
   },
   "training": {},
   "version": "0.1"