added model caching

UpstreetAI · Nov 28, 2023 · 11a1494 · 11a1494
1 parent bcf38c7
commit 11a1494
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -83,7 +83,10 @@ RUN mv '179257' /stable-diffusion-webui/models/Lora/gameiconlora.safetensors
 RUN wget https://civitai.com/api/download/models/96318
 RUN mv '96318' /stable-diffusion-webui/models/Lora/niji3d.safetensors
 
-RUN cd /stable-diffusion-webui && python cache.py --use-cpu=all --ckpt /model.safetensors
+RUN cd /stable-diffusion-webui && python cache.py --use-cpu=all --ckpt /stable-diffusion-webui/models/Lora/niji3d.safetensors
+RUN cd /stable-diffusion-webui && python cache.py --use-cpu=all --ckpt /stable-diffusion-webui/models/Lora/gameiconlora.safetensors
+RUN cd /stable-diffusion-webui && python cache.py --use-cpu=all --ckpt /stable-diffusion-webui/models/Stable-diffusion/niji.safetensors
+
 
 # Cleanup section (Worker Template)
 RUN apt-get autoremove -y && \