reaperdoesntknow commited on
Commit
fc7a930
·
verified ·
1 Parent(s): 44c77d1

Update config.json

Browse files
Files changed (1) hide show
  1. 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": null,
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": {},