Final config: Link correct multi-repo model sources

README.md

@@ -1,13 +1,11 @@
 ---
 title: Wan LoRA Trainer
 sdk: docker
-# This links all four required model repositories.
+# This links the TWO required model repositories.
 # Each one will be mounted as a separate folder inside the Space.
 models:
-- wan-video/wan2.1-i2v-14B-fp8-720p
-- wan-video/wan2.1-vae
-- wan-video/wan2.1-clip-xlm-roberta
-- wan-video/wan2.1-t5-xxl
+- Wan-AI/Wan2.1-I2V-14B-720P
+- Comfy-Org/Wan_2.1_ComfyUI_repackaged
 ---
 
 # Wan 2.1 LoRA Trainer

start.sh

@@ -1,43 +1,48 @@
 #!/bin/bash
-# --- Final Definitive Startup Script (v11) ---
+# --- Final Definitive Startup Script (v13 - Correct Multi-Repo Paths) ---
 
 # Exit immediately if a command exits with a non-zero status.
 set -e
 
 echo "--- Startup Script Initialized ---"
-echo "--- Models are mounted by the Space from multiple repositories. ---"
+echo "--- Models are mounted from two separate repositories. ---"
 
-# --- Define the correct paths for each mounted model repository ---
-DIT_DIR="/wan2.1-i2v-14B-fp8-720p"
-VAE_DIR="/wan2.1-vae"
-CLIP_DIR="/wan2.1-clip-xlm-roberta"
-T5_DIR="/wan2.1-t5-xxl"
+# --- Define the mount points for each repository ---
+# These names are automatically created by Hugging Face based on the repo names.
+WAN_AI_REPO_DIR="/Wan2.1-I2V-14B-720P"
+COMFY_REPO_DIR="/Wan_2.1_ComfyUI_repackaged"
 OUTPUT_DIR="/data/output"
 
-echo "DiT Path: $DIT_DIR"
-echo "VAE Path: $VAE_DIR"
-echo "CLIP Path: $CLIP_DIR"
-echo "T5 Path: $T5_DIR"
+# --- Define the full, correct paths for each model component ---
+DIT_PATH="$COMFY_REPO_DIR/split_files/diffusion_models/wan2.1_i2v_720p_14B_fp8_e4m3fn.safetensors"
+VAE_PATH="$COMFY_REPO_DIR/split_files/vae/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"
+
+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"
 
-# For robust verification, check for the existence of one file from each repo
-if [ ! -f "$DIT_DIR/wan2.1_i2v_720p_14B_fp8_e4m3fn.safetensors" ]; then
-    echo "CRITICAL ERROR: DiT model not found. Check README.md linking for 'wan-video/wan2.1-i2v-14B-fp8-720p'."
+# Verify that key files from both repositories exist
+if [ ! -f "$DIT_PATH" ]; then
+    echo "CRITICAL ERROR: DiT model not found. Check README.md linking for 'Comfy-Org/Wan_2.1_ComfyUI_repackaged'."
     exit 1
 fi
-if [ ! -f "$VAE_DIR/Wan2.1_VAE.pth" ]; then
-    echo "CRITICAL ERROR: VAE model not found. Check README.md linking for 'wan-video/wan2.1-vae'."
+if [ ! -f "$T5_PATH" ]; then
+    echo "CRITICAL ERROR: T5 model not found. Check README.md linking for 'Wan-AI/Wan2.1-I2V-14B-720P'."
     exit 1
 fi
 
 echo "All model repositories appear to be linked correctly. Starting training..."
-# Run the training command with the correct paths
+# Run the training command with the correct paths from each repository
 accelerate launch wan_train_network.py \
      --task "i2v-14B" \
-     --dit "$DIT_DIR/wan2.1_i2v_720p_14B_fp8_e4m3fn.safetensors" \
-     --vae "$VAE_DIR/Wan2.1_VAE.pth" \
-     --clip "$CLIP_DIR/models_clip_open-clip-xlm-roberta-large-vit-huge-14.pth" \
-     --t5 "$T5_DIR/models_t5_umt5-xxl-enc-bf16.pth" \
+     --dit "$DIT_PATH" \
+     --vae "$VAE_PATH" \
+     --clip "$CLIP_PATH" \
+     --t5 "$T5_PATH" \
      --dataset_config "dataset/testtoml.toml" \
      --output_dir "$OUTPUT_DIR" \
      --output_name "My_HF_Lora_v1" \