Update download_models.sh

download_models.sh

@@ -1,39 +1,24 @@
 #!/bin/bash
 
-# Function to download a file if it doesn't exist
-download_file() {
-    local url=$1
-    local target_dir=$2
-    local target_file=$3
+# Configure these variables
+COMFYUI_DIR="/workspace/ComfyUI"
 
-    if [ ! -f "${target_file}" ]; then
-        echo "Downloading ${target_file}..."
-        mkdir -p "${target_dir}"
-        wget -O "${target_file}" "${url}" --progress=bar:force:noscroll
-    else
-        echo "${target_file} already exists, skipping download."
-    fi
-}
+# Create required directories
+mkdir -p ${COMFYUI_DIR}/models/{diffusion_models,vae,clip}
 
+# Download models only if they don't exist
+wget -nc -P ${COMFYUI_DIR}/models/diffusion_models \
+  --header="Authorization: Bearer ${HF_TOKEN}" \
+  "https://huggingface.co/black-forest-labs/FLUX.1-dev/resolve/main/flux1-dev.safetensors"
 
-# VAE model
-download_file "https://huggingface.co/black-forest-labs/FLUX.1-dev/resolve/main/ae.sft?download=true" \
-              "/ComfyUI/models/vae" \
-              "/ComfyUI/models/vae/ae.sft"
+wget -nc -P ${COMFYUI_DIR}/models/vae \
+  --header="Authorization: Bearer ${HF_TOKEN}" \
+  "https://huggingface.co/black-forest-labs/FLUX.1-schnell/resolve/main/ae.safetensors"
 
-# Diffusion model
-download_file "https://huggingface.co/black-forest-labs/FLUX.1-dev/resolve/main/flux1-dev.sft?download=true" \
-              "/ComfyUI/models/diffusion_models" \
-              "/ComfyUI/models/diffusion_models/flux1-dev.sft"
+wget -nc -P ${COMFYUI_DIR}/models/clip \
+  --header="Authorization: Bearer ${HF_TOKEN}" \
+  "https://huggingface.co/comfyanonymous/flux_text_encoders/resolve/main/clip_l.safetensors"
 
-# CLIP models
-download_file "https://huggingface.co/comfyanonymous/flux_text_encoders/resolve/main/clip_l.safetensors?download=true" \
-              "/ComfyUI/models/clip" \
-              "/ComfyUI/models/clip/clip_l.safetensors"
-
-download_file "https://huggingface.co/comfyanonymous/flux_text_encoders/resolve/main/t5xxl_fp8_e4m3fn.safetensors?download=true" \
-              "/ComfyUI/models/clip" \
-              "/ComfyUI/models/clip/t5xxl_fp8_e4m3fn.safetensors"
-
-
-echo "All required files have been checked/Downloaded!"
+wget -nc -P ${COMFYUI_DIR}/models/clip \
+  --header="Authorization: Bearer ${HF_TOKEN}" \
+  "https://huggingface.co/comfyanonymous/flux_text_encoders/resolve/main/t5xxl_fp16.safetensors"