mattbusi commited on
Commit
6e4b11e
·
verified ·
1 Parent(s): 905bbee

config: restore torch_dtype for transformers<5 consumers

Browse files
Files changed (1) hide show
  1. config.json +3 -2
config.json CHANGED
@@ -272,5 +272,6 @@
272
  "num_attention_heads": 16,
273
  "num_hidden_layers": 27,
274
  "patch_size": 14
275
- }
276
- }
 
 
272
  "num_attention_heads": 16,
273
  "num_hidden_layers": 27,
274
  "patch_size": 14
275
+ },
276
+ "torch_dtype": "bfloat16"
277
+ }