Update provision.sh

provision.sh

@@ -26,6 +26,8 @@ echo "Checking Hugging Face models..."
 [ ! -f /workspace/ComfyUI/models/vae/flux_vae.safetensors ] && wget -P /workspace/ComfyUI/models/vae https://huggingface.co/StableDiffusionVN/Flux/resolve/main/Vae/flux_vae.safetensors
 [ ! -f /workspace/ComfyUI/models/ultralytics/bbox/FacesV1.pt ] && wget -P /workspace/ComfyUI/models/ultralytics/bbox "https://huggingface.co/ashllay/YOLO_Models/resolve/e07b01219ff1807e1885015f439d788b038f49bd/bbox/FacesV1.pt"
 [ ! -f /workspace/ComfyUI/models/upscale_models/1x-ITF-SkinDiffDetail-Lite-v1.pth ] && wget -P /workspace/ComfyUI/models/upscale_models/ "https://huggingface.co/cozyga/test/resolve/main/1x-ITF-SkinDiffDetail-Lite-v1.pth"
+[ ! -f /workspace/ComfyUI/models/checkpoints/LustifyENDGAME.safetensors ] && curl -L -H "$AUTH_HEADER" -o /workspace/ComfyUI/models/checkpoints/LustifyENDGAME.safetensors "https://civitai.com/api/download/models/1094291?type=Model&format=SafeTensor&size=pruned&fp=fp16"
+[ ! -f /workspace/ComfyUI/models/checkpoints/analogMadnessXL.safetensors ] && curl -L -H "$AUTH_HEADER" -o /workspace/ComfyUI/models/checkpoints/analogMadnessXL.safetensors "https://civitai.com/api/download/models/2207703?type=Model&format=SafeTensor&size=full&fp=fp16"
 
 # LoRAs (Hugging Face)
 LORA_DIR="/workspace/ComfyUI/models/loras"
@@ -40,7 +42,7 @@ LORA_DIR="/workspace/ComfyUI/models/loras"
 [ ! -f "$LORA_DIR/s0fia_v2_000004800.safetensors" ] && wget -P "$LORA_DIR" https://huggingface.co/cozyga/test/resolve/main/s0fia_v2_000004800.safetensors
 [ ! -f "$LORA_DIR/s0fia_v2_000003700.safetensors" ] && wget -P "$LORA_DIR" https://huggingface.co/cozyga/test/resolve/main/s0fia_v2_000003700.safetensors
 [ ! -f "$LORA_DIR/s0fia_v2_000004300.safetensors" ] && wget -P "$LORA_DIR" https://huggingface.co/cozyga/test/resolve/main/s0fia_v2_000004300.safetensors
-
+[ ! -f "$LORA_DIR/dmd2_sdxl_4step_lora_fp16.safetensors" ] && wget -P "$LORA_DIR" https://huggingface.co/tianweiy/DMD2/resolve/main/dmd2_sdxl_4step_lora_fp16.safetensors
 # 4. === MODEL DOWNLOADS (Civitai) ===
 echo "Checking Civitai models..."
 
@@ -57,6 +59,7 @@ echo "Checking Civitai models..."
 [ ! -f /workspace/ComfyUI/models/loras/nextscene.safetensors ] && \
     curl -L -H "$AUTH_HEADER" -o /workspace/ComfyUI/models/loras/nextscene.safetensors "https://civitai.com/api/download/models/2337214?type=Model&format=SafeTensor"
 
+
 # 5. === UPDATE AND START ===
 echo "Provisioning complete. Updating ComfyUI..."