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

start.sh

@@ -1,55 +1,46 @@
 #!/bin/bash
-# --- Final Definitive Startup Script (v23.2 - Uses USER-VERIFIED filenames) ---
+# --- Final Definitive Startup Script (v23.2 - Assumes Persistent Storage is enabled) ---
 
 set -e
 echo "--- Startup Script Initialized ---"
 
-# --- Define where models will be downloaded ---
+# This path is correct ONLY if persistent storage is enabled at /data
 MODELS_DIR="/data/models"
-mkdir -p $MODELS_DIR # Create the directory if it doesn't exist
+OUTPUT_DIR="/data/output"
+
+# Create our subdirectories inside the persistent storage
+mkdir -p $MODELS_DIR
+mkdir -p $OUTPUT_DIR
 
-# --- Define paths to the model files with CORRECTED names ---
+# --- Define paths to the model files ---
 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"
+CLIP_PATH="$MODELS_DIR/models_clip_open-clip-xlm-roberta-large-vit-huge-14.pth"
+T5_PATH="$MODELS_DIR/models_t5_umt5-xxl-enc-bf16.pth"
 
 # --- Download models from our dedicated model repo IF they don't already exist ---
-echo "--- Checking for model files... ---"
+echo "--- Checking for model files in persistent storage... ---"
 
 if [ ! -f "$DIT_PATH" ]; then
-    echo "Downloading DiT model from jujutechnology/WANfortraining..."
+    echo "Downloading DiT model..."
     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
-
+# ... (rest of the download checks are the same) ...
 if [ ! -f "$VAE_PATH" ]; then
-    echo "Downloading VAE model from jujutechnology/WANfortraining..."
+    echo "Downloading VAE model..."
     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
-
 if [ ! -f "$CLIP_PATH" ]; then
-    echo "Downloading CLIP model from jujutechnology/WANfortraining..."
-    # CORRECTED FILENAME in download command
+    echo "Downloading CLIP model..."
     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 "Downloading T5 model from jujutechnology/WANfortraining..."
-    # CORRECTED FILENAME in download command
+    echo "Downloading T5 model..."
     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 models are present. Starting training... ---"
-ls -lh $MODELS_DIR # List the downloaded files to confirm
+ls -lh $MODELS_DIR
 
 # --- Run the training command ---
 accelerate launch wan_train_network.py \