issues
search
christophschuhmann
/
improved-aesthetic-predictor
CLIP+MLP Aesthetic Score Predictor
Apache License 2.0
897
stars
88
forks
source link
Added requirements.txt and CPU only inference option
#10
Open
varna9000
opened
1 year ago
varna9000
commented
1 year ago
Created requirements.txt for easier pip install
Added checks for CPU/CUDA in simple_inference.py, so we run the model also on CPU only systems. Tested on Macbook Pro M1 with python 3.11