Update start.sh to use correct model filenames from dedicated repo

start.sh

@@ -1,45 +1,57 @@
 #!/bin/bash
+# --- Final Definitive Startup Script (v23.2 - Uses USER-VERIFIED filenames) ---
 
-# --- Final Definitive Startup Script (v21 - Points directly to the single DiT file) ---
-
-# Exit immediately if a command exits with a non-zero status.
 set -e
-
 echo "--- Startup Script Initialized ---"
-echo "--- Models are mounted from two separate repositories. ---"
 
-# --- Define the mount points for each repository ---
-WAN_AI_REPO_DIR="/Wan-AI/Wan2.1-I2V-14B-720P"
-COMFY_REPO_DIR="/Comfy-Org/Wan_2.1_ComfyUI_repackaged"
+# --- Define where models will be downloaded ---
+MODELS_DIR="/data/models"
+mkdir -p $MODELS_DIR # Create the directory if it doesn't exist
+
+# --- Define paths to the model files with CORRECTED names ---
+DIT_PATH="$MODELS_DIR/wan2.1_i2v_720p_14B_fp8_e4m3fn.safetensors"
+VAE_PATH="$MODELS_DIR/Wan2.1_VAE.pth"
+CLIP_PATH="$MODELS_DIR/models_clip_open-clip-xlm-roberta-large-vit-huge-14.pth" # CORRECTED FILENAME
+T5_PATH="$MODELS_DIR/models_t5_umt5-xxl-enc-bf16.pth" # CORRECTED FILENAME
 OUTPUT_DIR="/data/output"
 
+# --- Download models from our dedicated model repo IF they don't already exist ---
+echo "--- Checking for model files... ---"
 
-# --- Define the full, correct paths for each model component ---
-# The DiT model is a single file, so we point directly to its location.
-DIT_PATH="${COMFY_REPO_DIR}/split_files/diffusion_models/wan2.1_i2v_720p_14B_fp8_e4m3fn.safetensors"
-VAE_PATH="${WAN_AI_REPO_DIR}/Wan2.1-VAE.pth"
-CLIP_PATH="${WAN_AI_REPO_DIR}/models/clip_open-clip-xlm-roberta-large-vit-huge-14.pth"
-T5_PATH="${WAN_AI_REPO_DIR}/models/t5_umt5-xxl-enc-bf16.pth"
+if [ ! -f "$DIT_PATH" ]; then
+    echo "Downloading DiT model from jujutechnology/WANfortraining..."
+    huggingface-cli download jujutechnology/WANfortraining wan2.1_i2v_720p_14B_fp8_e4m3fn.safetensors --local-dir $MODELS_DIR --local-dir-use-symlinks False
+else
+    echo "DiT model already exists."
+fi
 
-echo "DiT Path: $DIT_PATH"
-echo "VAE Path: $VAE_PATH"
-echo "CLIP Path: $CLIP_PATH"
-echo "T5 Path: $T5_PATH"
-echo "Output Path: $OUTPUT_DIR"
+if [ ! -f "$VAE_PATH" ]; then
+    echo "Downloading VAE model from jujutechnology/WANfortraining..."
+    huggingface-cli download jujutechnology/WANfortraining Wan2.1_VAE.pth --local-dir $MODELS_DIR --local-dir-use-symlinks False
+else
+    echo "VAE model already exists."
+fi
 
-# --- Verify that key files from both repositories exist ---
-if [ ! -f "$DIT_PATH" ]; then
-    echo "CRITICAL ERROR: DiT model not found. Check the path in start.sh and the 'Comfy-Org/Wan_2.1_ComfyUI_repackaged' repository link."
-    exit 1
+if [ ! -f "$CLIP_PATH" ]; then
+    echo "Downloading CLIP model from jujutechnology/WANfortraining..."
+    # CORRECTED FILENAME in download command
+    huggingface-cli download jujutechnology/WANfortraining models_clip_open-clip-xlm-roberta-large-vit-huge-14.pth --local-dir $MODELS_DIR --local-dir-use-symlinks False
+else
+    echo "CLIP model already exists."
 fi
 
 if [ ! -f "$T5_PATH" ]; then
-    echo "CRITICAL ERROR: T5 model not found. Check the path in start.sh and the 'Wan-AI/Wan2.1-I2V-14B-720P' repository link."
-    exit 1
+    echo "Downloading T5 model from jujutechnology/WANfortraining..."
+    # CORRECTED FILENAME in download command
+    huggingface-cli download jujutechnology/WANfortraining models_t5_umt5-xxl-enc-bf16.pth --local-dir $MODELS_DIR --local-dir-use-symlinks False
+else
+    echo "T5 model already exists."
 fi
 
-echo "--- All model repositories appear to be linked correctly. Starting training... ---"
-# --- Run the training command with the correct paths from each repository ---
+echo "--- All models are present. Starting training... ---"
+ls -lh $MODELS_DIR # List the downloaded files to confirm
+
+# --- Run the training command ---
 accelerate launch wan_train_network.py \
     --task="i2v-lora" \
     --dit="$DIT_PATH" \