mattbusi commited on
Commit
e4bf432
·
verified ·
1 Parent(s): 57f9a1b

config: restore torch_dtype for transformers<5 consumers

Browse files
Files changed (1) hide show
  1. config.json +2 -1
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
  }