aiwithoutborders-xyz
/

CommunityForensics-DeepfakeDet-ViT

Image Classification

deepfake_detection

Model card Files Files and versions

LPX55 commited on Mar 13

Commit

afd376a

·

verified ·

1 Parent(s): 477be33

Update config.json

Files changed (1) hide show

config.json +2 -5

config.json CHANGED Viewed

@@ -1,11 +1,9 @@
 {
   "architectures": [
-    "ViTForImageClassification",
-    "ViT",
-    "Transformers"
   ],
   "model": {
-    "type": "vit",
     "variant": "vit_small_patch16_384.augreg_in21k_ft_in1k",
     "input_size": 384,
     "patch_size": 16,
@@ -25,7 +23,6 @@
     "resize_size": 440,
     "crop_size": 384
   },
-  "device": "cuda",
   "dtype": "float32",
   "checkpoint_path": "pretrained_weights/model_v11_ViT_384_base_ckpt.pt"
 }

 {
   "architectures": [
+    "ViTForImageClassification"
   ],
   "model": {
+    "type": "timm_wrapper",
     "variant": "vit_small_patch16_384.augreg_in21k_ft_in1k",
     "input_size": 384,
     "patch_size": 16,
     "resize_size": 440,
     "crop_size": 384
   },
   "dtype": "float32",
   "checkpoint_path": "pretrained_weights/model_v11_ViT_384_base_ckpt.pt"
 }