cozyga commited on
Commit
87d85d7
·
verified ·
1 Parent(s): 96c7ab4

Update cozywf

Browse files
Files changed (1) hide show
  1. cozywf +68 -0
cozywf CHANGED
@@ -0,0 +1,68 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+
3
+ # Vast.ai ComfyUI Provisioning Script
4
+ # Updated: Includes update command and model verification
5
+
6
+ echo "Starting provisioning script..."
7
+
8
+ # 1. Ensure all required directories exist
9
+ mkdir -p /workspace/ComfyUI/models/{text_encoders,checkpoints,vae,clip,loras,diffusion_models,upscale_models}
10
+ mkdir -p /workspace/ComfyUI/models/ultralytics/bbox
11
+
12
+ # 2. Define Helper for Civitai Downloads
13
+ AUTH_HEADER="Authorization: Bearer $CIVITAI_TOKEN"
14
+
15
+ # 3. === MODEL DOWNLOADS (Hugging Face) ===
16
+ echo "Checking Hugging Face models..."
17
+
18
+ # Flux & Qwen Base Models
19
+ [ ! -f /workspace/ComfyUI/models/text_encoders/t5xxl_fp8_e4m3fn_scaled.safetensors ] && wget -P /workspace/ComfyUI/models/text_encoders https://huggingface.co/comfyanonymous/flux_text_encoders/resolve/main/t5xxl_fp8_e4m3fn_scaled.safetensors
20
+ [ ! -f /workspace/ComfyUI/models/checkpoints/Qwen-Rapid-AIO-SFW-v23.safetensors ] && wget -P /workspace/ComfyUI/models/checkpoints https://huggingface.co/Phr00t/Qwen-Image-Edit-Rapid-AIO/resolve/main/v23/Qwen-Rapid-AIO-SFW-v23.safetensors
21
+ [ ! -f /workspace/ComfyUI/models/checkpoints/Qwen-Rapid-AIO-NSFW-v23.safetensors ] && wget -P /workspace/ComfyUI/models/checkpoints https://huggingface.co/Phr00t/Qwen-Image-Edit-Rapid-AIO/resolve/main/v23/Qwen-Rapid-AIO-NSFW-v23.safetensors
22
+ [ ! -f /workspace/ComfyUI/models/vae/qwen_image_vae.safetensors ] && wget -P /workspace/ComfyUI/models/vae "https://huggingface.co/Comfy-Org/Qwen-Image_ComfyUI/resolve/main/split_files/vae/qwen_image_vae.safetensors"
23
+ [ ! -f /workspace/ComfyUI/models/clip/qwen_3_4b.safetensors ] && wget -P /workspace/ComfyUI/models/clip https://huggingface.co/Comfy-Org/z_image_turbo/resolve/main/split_files/text_encoders/qwen_3_4b.safetensors
24
+ [ ! -f /workspace/ComfyUI/models/text_encoders/qwen_2.5_vl_7b_fp8_scaled.safetensors ] && wget -P /workspace/ComfyUI/models/text_encoders "https://huggingface.co/Comfy-Org/Qwen-Image_ComfyUI/resolve/main/split_files/text_encoders/qwen_2.5_vl_7b_fp8_scaled.safetensors"
25
+ [ ! -f /workspace/ComfyUI/models/diffusion_models/z_image_turbo_bf16.safetensors ] && wget -P /workspace/ComfyUI/models/diffusion_models https://huggingface.co/Comfy-Org/z_image_turbo/resolve/main/split_files/diffusion_models/z_image_turbo_bf16.safetensors
26
+ [ ! -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
27
+ [ ! -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"
28
+ [ ! -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"
29
+
30
+ # LoRAs (Hugging Face)
31
+ LORA_DIR="/workspace/ComfyUI/models/loras"
32
+ [ ! -f "$LORA_DIR/Z-Isla_3K_0002_000003500.safetensors" ] && wget -P "$LORA_DIR" https://huggingface.co/cozyga/test/resolve/main/Z-Isla_3K_0002_000003500.safetensors
33
+ [ ! -f "$LORA_DIR/z-kyla-v1.safetensors" ] && wget -P "$LORA_DIR" https://huggingface.co/cozyga/test/resolve/main/z-kyla-v1.safetensors
34
+ [ ! -f "$LORA_DIR/m1a_zit_3.safetensors" ] && wget -P "$LORA_DIR" https://huggingface.co/cozyga/test/resolve/main/m1a_zit_3.safetensors
35
+ [ ! -f "$LORA_DIR/c3lia_v1_000003600.safetensors" ] && wget -P "$LORA_DIR" https://huggingface.co/cozyga/test/resolve/main/c3lia_v1_000003600.safetensors
36
+ [ ! -f "$LORA_DIR/c3lia_v1_000003900.safetensors" ] && wget -P "$LORA_DIR" https://huggingface.co/cozyga/test/resolve/main/c3lia_v1_000003900.safetensors
37
+ [ ! -f "$LORA_DIR/qwen_c3lia_000003600.safetensors" ] && wget -P "$LORA_DIR" https://huggingface.co/cozyga/test/resolve/main/qwen_c3lia_000003600.safetensors
38
+ [ ! -f "$LORA_DIR/l0ki_zit_v1_000003000.safetensors" ] && wget -P "$LORA_DIR" https://huggingface.co/XLB666/qwenzitloras/resolve/main/l0ki_zit_v1_000003000.safetensors
39
+ [ ! -f "$LORA_DIR/l0ki_qwen_v1_000002800.safetensors" ] && wget -P "$LORA_DIR" https://huggingface.co/XLB666/qwenzitloras/resolve/main/l0ki_qwen_v1_000002800.safetensors
40
+ [ ! -f "$LORA_DIR/s0fia_v2_000004800.safetensors" ] && wget -P "$LORA_DIR" https://huggingface.co/cozyga/test/resolve/main/s0fia_v2_000004800.safetensors
41
+ [ ! -f "$LORA_DIR/s0fia_v2_000003700.safetensors" ] && wget -P "$LORA_DIR" https://huggingface.co/cozyga/test/resolve/main/s0fia_v2_000003700.safetensors
42
+ [ ! -f "$LORA_DIR/s0fia_v2_000004300.safetensors" ] && wget -P "$LORA_DIR" https://huggingface.co/cozyga/test/resolve/main/s0fia_v2_000004300.safetensors
43
+
44
+ # 4. === MODEL DOWNLOADS (Civitai) ===
45
+ echo "Checking Civitai models..."
46
+
47
+ [ ! -f /workspace/ComfyUI/models/loras/zittitslider.safetensors ] && \
48
+ curl -L -H "$AUTH_HEADER" -o /workspace/ComfyUI/models/loras/zittitslider.safetensors "https://civitai.com/api/download/models/2478366"
49
+ [ ! -f /workspace/ComfyUI/models/loras/QwenEditVersatilePoses.safetensors ] && \
50
+ curl -L -H "$AUTH_HEADER" -o /workspace/ComfyUI/models/loras/QwenEditVersatilePoses.safetensors "https://civitai.com/api/download/models/2457989?type=Model&format=SafeTensor"
51
+ [ ! -f /workspace/ComfyUI/models/loras/sdxlcloseuppussy.safetensors ] && \
52
+ curl -L -H "$AUTH_HEADER" -o /workspace/ComfyUI/models/loras/sdxlcloseuppussy.safetensors "https://civitai.com/api/download/models/2376235?type=Model&format=SafeTensor"
53
+ [ ! -f /workspace/ComfyUI/models/vae/zImage_vae.safetensors ] && \
54
+ curl -L -H "$AUTH_HEADER" -o /workspace/ComfyUI/models/vae/zImage_vae.safetensors "https://civitai.com/api/download/models/2442479"
55
+ [ ! -f /workspace/ComfyUI/models/loras/amateurphotography.safetensors ] && \
56
+ curl -L -H "$AUTH_HEADER" -o /workspace/ComfyUI/models/loras/amateurphotography.safetensors "https://civitai.com/api/download/models/2363467"
57
+ [ ! -f /workspace/ComfyUI/models/loras/nextscene.safetensors ] && \
58
+ curl -L -H "$AUTH_HEADER" -o /workspace/ComfyUI/models/loras/nextscene.safetensors "https://civitai.com/api/download/models/2337214?type=Model&format=SafeTensor"
59
+
60
+ # 5. === UPDATE AND START ===
61
+ echo "Provisioning complete. Updating ComfyUI..."
62
+
63
+ cd /workspace/ComfyUI || exit
64
+ git pull
65
+
66
+ echo "ComfyUI is updated. Starting now..."
67
+ # Ensure the script ends by handing over to the ComfyUI process
68
+ python main.py --listen