akagtag commited on
Commit
10580aa
Β·
1 Parent(s): a38026a

refactor: clean up model roster comments and remove broken model references

Browse files
Files changed (2) hide show
  1. FIX.md +12 -18
  2. src/engines/fingerprint/engine.py +1 -14
FIX.md CHANGED
@@ -13,13 +13,8 @@ Copy `fingerprint_engine.py` (from outputs) to:
13
  src/engines/fingerprint/engine.py
14
  ```
15
 
16
- This removes ALL broken models:
17
-
18
- - `yermandy/deepfake-detection` β€” gone
19
- - `yermandy/GenD_CLIP_L_14` β€” gone
20
- - `yermandy/GenD_DINOv3_L` β€” gone
21
- - `Wvolf/ViT_Deepfake_Detection` β€” gone
22
- - `trust_remote_code` kwarg bug β€” fixed
23
 
24
  Replaces with 3 working models:
25
 
@@ -37,10 +32,9 @@ Copy `coherence_engine.py` (from outputs) to:
37
  src/engines/coherence/engine.py
38
  ```
39
 
40
- This removes the broken wav2vec model
41
- (`nii-yamagishilab/wav2vec-large-anti-deepfake-nda`) which has incompatible
42
- weights and was producing random output anyway. Coherence now runs visual-only
43
- (FaceNet + MediaPipe).
44
 
45
  ---
46
 
@@ -52,7 +46,8 @@ Copy `sstgnn_engine.py` (from outputs) to:
52
  src/engines/sstgnn/engine.py
53
  ```
54
 
55
- Removes `Wvolf/ViT_Deepfake_Detection`. Uses `dima806` + `prithivMLmods` only.
 
56
 
57
  ---
58
 
@@ -110,10 +105,9 @@ torchvision>=0.21.0
110
  torchaudio>=2.6.0
111
  ```
112
 
113
- Torch < 2.6 blocks loading `.pt` files due to CVE-2025-32434.
114
- `Wvolf/ViT_Deepfake_Detection` uses `.pt` β€” it will NEVER load on torch < 2.6.
115
- Since you're removing that model anyway, this is a safety measure for other
116
- models.
117
 
118
  ---
119
 
@@ -167,8 +161,8 @@ SSTGNN engine ready: 2 detectors, mediapipe=ok
167
 
168
  | Error | Cause | Fix |
169
  | --------------------------------- | ------------------------------------------- | --------------------------------- |
170
- | `yermandy/*` warnings | custom GenD arch | removed from engine |
171
- | `Wvolf/*` torch CVE error | .pt file + torch < 2.6 | removed from engine |
172
  | `trust_remote_code` TypeError | duplicate kwarg in \_build_image_classifier | removed from all pipeline() calls |
173
  | `wav2vec` MISSING/UNEXPECTED keys | custom m_ssl.\* namespace, incompatible | removed from engine |
174
  | `libGLESv2.so.2 not found` | missing apt packages in Docker | add libgles2 libegl1 libgbm1 |
 
13
  src/engines/fingerprint/engine.py
14
  ```
15
 
16
+ This removes all broken legacy models and fixes the `trust_remote_code`
17
+ duplicate kwarg bug.
 
 
 
 
 
18
 
19
  Replaces with 3 working models:
20
 
 
32
  src/engines/coherence/engine.py
33
  ```
34
 
35
+ This removes the broken wav2vec-based audio model path with incompatible
36
+ weights and random output behavior. Coherence now runs visual-only (FaceNet +
37
+ MediaPipe).
 
38
 
39
  ---
40
 
 
46
  src/engines/sstgnn/engine.py
47
  ```
48
 
49
+ Removes legacy ViT `.pt` checkpoint usage. Uses `dima806` + `prithivMLmods`
50
+ only.
51
 
52
  ---
53
 
 
105
  torchaudio>=2.6.0
106
  ```
107
 
108
+ Torch < 2.6 blocks loading `.pt` files due to CVE-2025-32434. Since legacy
109
+ `.pt` checkpoints are removed, this is still a safety measure for other
110
+ dependencies that may call `torch.load`.
 
111
 
112
  ---
113
 
 
161
 
162
  | Error | Cause | Fix |
163
  | --------------------------------- | ------------------------------------------- | --------------------------------- |
164
+ | Legacy GenD warnings | custom architecture | removed from engine |
165
+ | Legacy ViT torch CVE error | `.pt` file + torch < 2.6 | removed from engine |
166
  | `trust_remote_code` TypeError | duplicate kwarg in \_build_image_classifier | removed from all pipeline() calls |
167
  | `wav2vec` MISSING/UNEXPECTED keys | custom m_ssl.\* namespace, incompatible | removed from engine |
168
  | `libGLESv2.so.2 not found` | missing apt packages in Docker | add libgles2 libegl1 libgbm1 |
src/engines/fingerprint/engine.py CHANGED
@@ -24,20 +24,7 @@ from src.types import EngineResult
24
  logger = logging.getLogger(__name__)
25
  CACHE = os.environ.get("MODEL_CACHE_DIR", "/tmp/models")
26
 
27
- # ---------------------------------------------------------------------------
28
- # Model roster β€” all confirmed safetensors, standard HF architectures
29
- # ---------------------------------------------------------------------------
30
- # Primary: Organika/sdxl-detector β€” ViT-L/14 fine-tuned on 2M real/AI images
31
- # Secondary: haywoodsloan/ai-image-detector-deploy β€” EfficientNet, robust
32
- # Tertiary: dima806/deepfake_vs_real_image_detection β€” ResNet50, 70k faces
33
- #
34
- # REMOVED (broken):
35
- # yermandy/deepfake-detection β€” missing model_type in config.json
36
- # yermandy/GenD_CLIP_L_14 β€” custom GenD arch, not in Transformers
37
- # yermandy/GenD_DINOv3_L β€” same GenD arch issue
38
- # Wvolf/ViT_Deepfake_Detection β€” .pt file + trust_remote_code conflict +
39
- # torch CVE-2025-32434 block
40
- # ---------------------------------------------------------------------------
41
  DETECTOR_CANDIDATES = [
42
  "Organika/sdxl-detector",
43
  "haywoodsloan/ai-image-detector-deploy",
 
24
  logger = logging.getLogger(__name__)
25
  CACHE = os.environ.get("MODEL_CACHE_DIR", "/tmp/models")
26
 
27
+
 
 
 
 
 
 
 
 
 
 
 
 
 
28
  DETECTOR_CANDIDATES = [
29
  "Organika/sdxl-detector",
30
  "haywoodsloan/ai-image-detector-deploy",