fix: load LAION V2 directly via transformers (trust_remote_code); drop stale wrapper; pin requirements

This commit is contained in:
Kareem Horstink
2026-08-23 16:27:36 +00:00
parent 027788707e
commit 62f65a2d8e
2 changed files with 26 additions and 21 deletions
+6 -6
View File
@@ -1,13 +1,13 @@
# Photo Judgers — Requirements
# Install in conda env: pip install -r requirements.txt
#
# simple-aesthetics-predictor automatically installs torch + transformers.
# We avoid pinning torchvision/torchaudio here to prevent version mismatch
# with the torch that simple-aesthetics-predictor resolves. Pillow is needed
# for image loading; tqdm for progress bars.
# 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.
simple-aesthetics-predictor
transformers
transformers>=4.40
torch>=2.1
torchvision
Pillow
tqdm
numpy