framasoft commited on
Commit
b5e8993
·
verified ·
1 Parent(s): e2360e8

Update 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": "jaranohaal/vit-base-violence-detection",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
@@ -18,5 +18,6 @@
18
  "num_labels": 2,
19
  "transformers_version": "4.41.0",
20
  "torch_dtype": "float32",
21
- "model_type": "vit"
 
22
  }
 
1
  {
2
+ "_name_or_path": "framasoft/vit-base-violence-detection",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
 
18
  "num_labels": 2,
19
  "transformers_version": "4.41.0",
20
  "torch_dtype": "float32",
21
+ "model_type": "vit",
22
+ "image_processor_type": "vit"
23
  }