TryingHard commited on
Commit
f8aef0f
·
verified ·
1 Parent(s): 3c646d9

Update config.json to support transformers 4.49.0

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -180,7 +180,7 @@
180
  "tokenizer_class": null,
181
  "top_k": 50,
182
  "top_p": 1.0,
183
- "torch_dtype": "float32",
184
  "torchscript": false,
185
  "typical_p": 1.0,
186
  "use_bfloat16": false,
 
180
  "tokenizer_class": null,
181
  "top_k": 50,
182
  "top_p": 1.0,
183
+ "torch_dtype": "bfloat16",
184
  "torchscript": false,
185
  "typical_p": 1.0,
186
  "use_bfloat16": false,