diff --git a/examples/wanvideo/model_training/lora/Wan2.1-Fun-V1.1-1.3B-Control-Camera.sh b/examples/wanvideo/model_training/lora/Wan2.1-Fun-V1.1-1.3B-Control-Camera.sh index 721d88f..2f809a4 100644 --- a/examples/wanvideo/model_training/lora/Wan2.1-Fun-V1.1-1.3B-Control-Camera.sh +++ b/examples/wanvideo/model_training/lora/Wan2.1-Fun-V1.1-1.3B-Control-Camera.sh @@ -8,7 +8,7 @@ accelerate launch examples/wanvideo/model_training/train.py \ --learning_rate 1e-5 \ --num_epochs 5 \ --remove_prefix_in_ckpt "pipe.dit." \ - --output_path "./models/train/Wan2.1-Fun-V1.1-1.3B-Control-Camera_full" \ + --output_path "./models/train/Wan2.1-Fun-V1.1-1.3B-Control-Camera_lora" \ --lora_base_model "dit" \ --lora_target_modules "q,k,v,o,ffn.0,ffn.2" \ --lora_rank 32 \ diff --git a/examples/wanvideo/model_training/lora/Wan2.1-Fun-V1.1-14B-Control-Camera.sh b/examples/wanvideo/model_training/lora/Wan2.1-Fun-V1.1-14B-Control-Camera.sh index 9fd30c4..176a05f 100644 --- a/examples/wanvideo/model_training/lora/Wan2.1-Fun-V1.1-14B-Control-Camera.sh +++ b/examples/wanvideo/model_training/lora/Wan2.1-Fun-V1.1-14B-Control-Camera.sh @@ -8,7 +8,7 @@ accelerate launch examples/wanvideo/model_training/train.py \ --learning_rate 1e-5 \ --num_epochs 5 \ --remove_prefix_in_ckpt "pipe.dit." \ - --output_path "./models/train/Wan2.1-Fun-V1.1-14B-Control-Camera_full" \ + --output_path "./models/train/Wan2.1-Fun-V1.1-14B-Control-Camera_lora" \ --lora_base_model "dit" \ --lora_target_modules "q,k,v,o,ffn.0,ffn.2" \ --lora_rank 32 \ diff --git a/examples/wanvideo/model_training/validate_full/Wan2.1-VACE-1.3B-Preview.py b/examples/wanvideo/model_training/validate_full/Wan2.1-VACE-1.3B-Preview.py index 7db26e0..a916745 100644 --- a/examples/wanvideo/model_training/validate_full/Wan2.1-VACE-1.3B-Preview.py +++ b/examples/wanvideo/model_training/validate_full/Wan2.1-VACE-1.3B-Preview.py @@ -13,7 +13,7 @@ pipe = WanVideoPipeline.from_pretrained( ModelConfig(model_id="iic/VACE-Wan2.1-1.3B-Preview", origin_file_pattern="Wan2.1_VAE.pth", offload_device="cpu"), ], ) -state_dict = load_state_dict("models/train/VACE-Wan2.1-1.3B-Preview_full/epoch-1.safetensors") +state_dict = load_state_dict("models/train/Wan2.1-VACE-1.3B-Preview_full/epoch-1.safetensors") pipe.vace.load_state_dict(state_dict) pipe.enable_vram_management()