Update start.sh

start.sh

@@ -40,45 +40,6 @@ fi
 # RUN COMFYUI
 python3 /workspace/ComfyUI/main.py --listen
 
-
-# Function to download a file if it doesn't exist
-download_file() {
-    local url=$1
-    local target_dir=$2
-    local target_file=$3
-
-    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
-}
-
-
-# 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"
-
-# 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"
-
-# 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!"
-
-
+/download_models.sh
 
 sleep infinity