Update config.json to support transformers 4.49.0
Browse files- 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": "
|
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,
|