cicdatopea commited on
Commit
8d67fa6
·
verified ·
1 Parent(s): 5e2ce33

Update config.json

Browse files

change to bf16 dtype

Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -78,7 +78,7 @@
78
  "tie_word_embeddings": false,
79
  "topk_group": 4,
80
  "topk_method": "noaux_tc",
81
- "torch_dtype": "float16",
82
  "transformers_version": "4.47.0",
83
  "use_cache": true,
84
  "v_head_dim": 128,
 
78
  "tie_word_embeddings": false,
79
  "topk_group": 4,
80
  "topk_method": "noaux_tc",
81
+ "torch_dtype": "bfloat16",
82
  "transformers_version": "4.47.0",
83
  "use_cache": true,
84
  "v_head_dim": 128,