Update start.sh to add arguments

start.sh

@@ -1,24 +1,20 @@
 #!/bin/bash
-# --- Final Definitive Startup Script (v23.6 - Points to the correct TOML file) ---
+# --- Final Definitive Startup Script (v23.7 - Re-adds necessary LoRA arguments) ---
 
 set -e
 echo "--- Startup Script Initialized ---"
 
-# This path is correct ONLY if persistent storage is enabled at /data
 MODELS_DIR="/data/models"
 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 ---
 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"
 T5_PATH="$MODELS_DIR/models_t5_umt5-xxl-enc-bf16.pth"
 
-# --- Download models if they don't exist ---
 echo "--- Checking for model files in persistent storage... ---"
 if [ ! -f "$DIT_PATH" ]; then
     echo "Downloading DiT model..."
@@ -40,7 +36,7 @@ fi
 echo "--- All models are present. Starting training... ---"
 ls -lh $MODELS_DIR
 
-# --- Run the training command with CORRECTED dataset config ---
+# --- Run the training command with CORRECT LoRA arguments ---
 accelerate launch wan_train_network.py \
     --task="i2v-14B" \
     --dit="$DIT_PATH" \
@@ -50,6 +46,9 @@ accelerate launch wan_train_network.py \
     --dataset_config="dataset/huggingfacetoml.toml" \
     --output_dir="$OUTPUT_DIR" \
     --output_name="my-I2V-Lora" \
+    --network_module="networks.lora_wan" \
+    --network_dim="32" \
+    --network_alpha="4" \
     --max_train_epochs="70" \
     --learning_rate="1e-5" \
     --optimizer_type="adamw" \