Update Dockerfile

Dockerfile

@@ -68,7 +68,8 @@ RUN echo "Downloading checkpoints..." && \
     # TechnoByte Models
     wget -cq https://huggingface.co/TechnoByte/MilkyWonderland/resolve/main/milkyWonderland_v40.safetensors -P ./models/checkpoints/ && \
     # LCM
-    wget -O ./models/loras/lcm-lora-sdv1-5.safetensors https://huggingface.co/latent-consistency/lcm-lora-sdv1-5/resolve/main/pytorch_lora_weights.safetensors && \
+    #wget -O ./models/loras/lcm-lora-sdv1-5.safetensors https://huggingface.co/latent-consistency/lcm-lora-sdv1-5/resolve/main/pytorch_lora_weights.safetensors && \
+    wget -O ./models/loras/v2-1_768-ema-pruned.safetensors https://huggingface.co/stabilityai/stable-diffusion-2-1/resolve/main/v2-1_768-ema-pruned.safetensors && \
     
     wget -O ./models/checkpoints/v1-5-pruned-emaonly.ckpt https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt && \
     
@@ -84,7 +85,8 @@ RUN echo "Downloading checkpoints..." && \
     wget -cq https://huggingface.co/lllyasviel/ControlNet-v1-1/resolve/main/control_v11p_sd15_openpose.pth -P ./models/controlnet/ && \
     # GLIGEN
     # wget -c https://huggingface.co/comfyanonymous/GLIGEN_pruned_safetensors/resolve/main/gligen_sd14_textbox_pruned_fp16.safetensors -P ./models/gligen/ && \ 
-
+    #vae
+    wget -O ./models/vae/vae-ft-mse-840000-ema-pruned.ckpt https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.ckpt && \
     # ComfyUI Manager
     cd custom_nodes && git clone https://github.com/ltdrdata/ComfyUI-Manager.git ComfyUI-Manager && \
     cd ComfyUI-Manager/glob && sed -i '/def is_allowed_security_level/,/return/s/return.*/return True/' manager_server.py && \