aznasut commited on
Commit
3007632
1 Parent(s): 4622998

change model

Browse files
Files changed (3) hide show
  1. config.json +7 -5
  2. model.safetensors +2 -2
  3. preprocessor_config.json +1 -1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "dima806/ai_vs_real_image_detection",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
@@ -9,15 +9,17 @@
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 768,
11
  "id2label": {
12
- "0": "REAL",
13
- "1": "FAKE"
 
14
  },
15
  "image_size": 224,
16
  "initializer_range": 0.02,
17
  "intermediate_size": 3072,
18
  "label2id": {
19
- "FAKE": 1,
20
- "REAL": 0
 
21
  },
22
  "layer_norm_eps": 1e-12,
23
  "model_type": "vit",
 
1
  {
2
+ "_name_or_path": "NYUAD-ComNets/AI-generated_images_detector",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
 
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 768,
11
  "id2label": {
12
+ "0": "dalle",
13
+ "1": "real",
14
+ "2": "sd"
15
  },
16
  "image_size": 224,
17
  "initializer_range": 0.02,
18
  "intermediate_size": 3072,
19
  "label2id": {
20
+ "dalle": "0",
21
+ "real": "1",
22
+ "sd": "2"
23
  },
24
  "layer_norm_eps": 1e-12,
25
  "model_type": "vit",
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4f5d0d4e8e6475faccca9385a2869648bb1f18d3f64c0239ad2bca909d3adc50
3
- size 343223968
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b0eb0ba578897f59d009b4eaa7a308eb5ca56a6c27e7d51a32d4eafcbe4a223a
3
+ size 343227052
preprocessor_config.json CHANGED
@@ -21,7 +21,7 @@
21
  0.5,
22
  0.5
23
  ],
24
- "image_processor_type": "ViTImageProcessor",
25
  "image_std": [
26
  0.5,
27
  0.5,
 
21
  0.5,
22
  0.5
23
  ],
24
+ "image_processor_type": "ViTFeatureExtractor",
25
  "image_std": [
26
  0.5,
27
  0.5,