fix: load LAION V2 directly via transformers (trust_remote_code); drop stale wrapper; pin requirements
This commit is contained in:
+6
-6
@@ -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
|
||||
Reference in New Issue
Block a user