Spaces:
Sleeping
Sleeping
refactor: clean up model roster comments and remove broken model references
Browse files- FIX.md +12 -18
- 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
|
| 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 |
-
|
| 42 |
-
|
| 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 `
|
|
|
|
| 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 |
-
`
|
| 115 |
-
|
| 116 |
-
models.
|
| 117 |
|
| 118 |
---
|
| 119 |
|
|
@@ -167,8 +161,8 @@ SSTGNN engine ready: 2 detectors, mediapipe=ok
|
|
| 167 |
|
| 168 |
| Error | Cause | Fix |
|
| 169 |
| --------------------------------- | ------------------------------------------- | --------------------------------- |
|
| 170 |
-
|
|
| 171 |
-
|
|
| 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",
|