TechnoByte commited on
Commit
d7041d5
·
1 Parent(s): f790a9f

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -0
Dockerfile CHANGED
@@ -72,6 +72,11 @@ RUN echo "Downloading checkpoints..." && \
72
  # ESRGAN upscale model
73
  wget -c https://huggingface.co/konohashinobi4/4xAnimesharp/resolve/main/4x-AnimeSharp.pth -P ./models/upscale_models/ && \
74
 
 
 
 
 
 
75
  # Install custom nodes
76
  echo "Installing custom nodes..."
77
 
 
72
  # ESRGAN upscale model
73
  wget -c https://huggingface.co/konohashinobi4/4xAnimesharp/resolve/main/4x-AnimeSharp.pth -P ./models/upscale_models/ && \
74
 
75
+ # Aesthetic scorer models
76
+ mkdir ./models/aesthetic && \
77
+ wget -c https://github.com/grexzen/SD-Chad/raw/main/chadscorer.pth -P ./models/aesthetic/ && \
78
+ wget -c https://github.com/christophschuhmann/improved-aesthetic-predictor/raw/main/ava+logos-l14-linearMSE.pth -P ./models/aesthetic/ && \
79
+
80
  # Install custom nodes
81
  echo "Installing custom nodes..."
82