ttangmo24 commited on
Commit
71bb791
1 Parent(s): 125d7ee

Upload ViTForImageClassification

Browse files
Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "google/vit-base-patch16-224-in21k",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
@@ -32,5 +32,5 @@
32
  "problem_type": "single_label_classification",
33
  "qkv_bias": true,
34
  "torch_dtype": "float32",
35
- "transformers_version": "4.41.1"
36
  }
 
1
  {
2
+ "_name_or_path": "ttangmo24/vit-base-classification-Eye-Diseases",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
 
32
  "problem_type": "single_label_classification",
33
  "qkv_bias": true,
34
  "torch_dtype": "float32",
35
+ "transformers_version": "4.41.2"
36
  }