Update start.sh

start.sh

@@ -36,6 +36,47 @@ if [ -d "/workspace/venv" ]; then
     source /workspace/venv/bin/activate
 fi
 
+
+# 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!"
+
+
+
+
 # Execute the user's script
 python3 /workspace/ComfyUI/main.py --listen