Vidit2003 commited on
Commit
791a96b
·
verified ·
1 Parent(s): 6c9bd17

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
- "architectures": ["vision_transformer"],
3
- "model_type": "vit",
4
  "num_channels": 1,
5
  "image_size": 224,
6
  "patch_size": 16,
 
1
  {
2
+ "architectures": ["VisionTransformer"],
3
+ "model_type": "vision_transformer",
4
  "num_channels": 1,
5
  "image_size": 224,
6
  "patch_size": 16,