aznasut
/

ai_vs_fake_image

Model card Files Files and versions

aznasut commited on Oct 29, 2024

Commit

3007632

·

1 Parent(s): 4622998

change model

Files changed (3) hide show

config.json +7 -5
model.safetensors +2 -2
preprocessor_config.json +1 -1

config.json CHANGED Viewed

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "dima806/ai_vs_real_image_detection",
   "architectures": [
     "ViTForImageClassification"
   ],
@@ -9,15 +9,17 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "REAL",
-    "1": "FAKE"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "FAKE": 1,
-    "REAL": 0
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",

 {
+  "_name_or_path": "NYUAD-ComNets/AI-generated_images_detector",
   "architectures": [
     "ViTForImageClassification"
   ],
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
+    "0": "dalle",
+    "1": "real",
+    "2": "sd"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
+    "dalle": "0",
+    "real": "1",
+    "sd": "2"
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",

model.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4f5d0d4e8e6475faccca9385a2869648bb1f18d3f64c0239ad2bca909d3adc50
-size 343223968

 version https://git-lfs.github.com/spec/v1
+oid sha256:b0eb0ba578897f59d009b4eaa7a308eb5ca56a6c27e7d51a32d4eafcbe4a223a
+size 343227052

preprocessor_config.json CHANGED Viewed

@@ -21,7 +21,7 @@
     0.5,
     0.5
   ],
-  "image_processor_type": "ViTImageProcessor",
   "image_std": [
     0.5,
     0.5,

     0.5,
     0.5
   ],
+  "image_processor_type": "ViTFeatureExtractor",
   "image_std": [
     0.5,
     0.5,