Fix lora config

Perseus14 · Perseus14 · commit fe9d3ed396a5 · 2026-01-21T14:41:41.000Z
diff --git a/src/maxdiffusion/configs/base_wan_lora_i2v_14b.yml b/src/maxdiffusion/configs/base_wan_lora_i2v_14b.yml
@@ -298,6 +298,7 @@ lightning_repo: ""
 lightning_ckpt: ""
 
 # LoRA parameters
+lora_rank: 64
 # Values are lists to support multiple LoRA loading during inference in the future.
 lora_config: {
   lora_model_name_or_path: ["lightx2v/Wan2.1-Distill-Loras"],
diff --git a/src/maxdiffusion/configs/base_wan_lora_i2v_27b.yml b/src/maxdiffusion/configs/base_wan_lora_i2v_27b.yml
@@ -310,6 +310,7 @@ lightning_repo: ""
 lightning_ckpt: ""
 
 # LoRA parameters
+lora_rank: 64
 # Values are lists to support multiple LoRA loading during inference in the future.
 lora_config: {
   lora_model_name_or_path: ["lightx2v/Wan2.2-Distill-Loras"],
