fix: self-contained LAION V2 (CLIP via transformers + official .pth head), no remote code, no stale wrapper

This commit is contained in:
Kareem Horstink
2026-08-23 17:31:13 +00:00
parent e60b4dce20
commit 016a68e0a2
2 changed files with 105 additions and 37 deletions
+6 -6
View File
@@ -1,13 +1,13 @@
# Photo Judgers — Requirements
# Install in conda env: pip install -r requirements.txt
#
# We load the LAION V2 model directly via transformers (trust_remote_code),
# so we do NOT depend on the stale `simple-aesthetics-predictor` wrapper
# (breaks with modern transformers). torch is required by transformers.
# LAION V2 scorer uses OpenAI CLIP via transformers (stable API, no remote
# code) plus a small linear head loaded from the official .pth. This avoids
# the stale `simple-aesthetics-predictor` wrapper and transformers remote
# code, both of which break on modern transformers.
transformers>=4.40
transformers>=4.30
torch>=2.1
torchvision
Pillow
tqdm
numpy
tqdm