Update config.json
Browse files- config.json +1 -1
config.json
CHANGED
@@ -19,7 +19,7 @@
|
|
19 |
"output_hidden_states": false,
|
20 |
"output_attentions": false,
|
21 |
"torchscript": false,
|
22 |
-
"torch_dtype":
|
23 |
"use_bfloat16": false,
|
24 |
"tf_legacy_loss": false,
|
25 |
"pruned_heads": {},
|
|
|
19 |
"output_hidden_states": false,
|
20 |
"output_attentions": false,
|
21 |
"torchscript": false,
|
22 |
+
"torch_dtype": "float16",
|
23 |
"use_bfloat16": false,
|
24 |
"tf_legacy_loss": false,
|
25 |
"pruned_heads": {},
|