config: restore torch_dtype for transformers<5 consumers
Browse files- 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 |
}
|