haiefff commited on
Commit
8f13fd3
1 Parent(s): 5eeb8a4

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +3 -2
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "model",
3
  "_num_labels": 2,
4
  "architectures": [
5
  "ViTForImageClassification"
@@ -28,5 +28,6 @@
28
  "patch_size": 16,
29
  "problem_type": "single_label_classification",
30
  "qkv_bias": true,
31
- "transformers_version": "4.33.2"
 
32
  }
 
1
  {
2
+ "_name_or_path": "google/vit-base-patch16-384",
3
  "_num_labels": 2,
4
  "architectures": [
5
  "ViTForImageClassification"
 
28
  "patch_size": 16,
29
  "problem_type": "single_label_classification",
30
  "qkv_bias": true,
31
+ "torch_dtype": "float32",
32
+ "transformers_version": "4.45.0"
33
  }