Fix bugs

Fix bugs

configs/wan22/wan_i2v_audio.json

@@ -20,12 +20,12 @@
     "use_image_encoder": false,
     "use_31_block": false,
     "lora_configs": [
-        { 
+        {
         "name": "high_noise_model",
         "path": "/mnt/Text2Video/wuzhuguanyu/Wan21_T2V_14B_lightx2v_cfg_step_distill_lora_rank64.safetensors",
         "strength": 1.0
         },
-        { 
+        {
         "name": "low_noise_model",
         "path": "/mnt/Text2Video/wuzhuguanyu/Wan21_T2V_14B_lightx2v_cfg_step_distill_lora_rank64.safetensors",
         "strength": 1.0

lightx2v/models/runners/default_runner.py

@@ -176,7 +176,7 @@ class DefaultRunner(BaseRunner):
             self.model = self.load_transformer()
         self.init_scheduler()
         self.model.scheduler.prepare(self.inputs["image_encoder_output"])
-        if self.config.get("model_cls") == "wan2.2":
+        if self.config.get("model_cls") == "wan2.2" and self.config["task"] == "i2v":
             self.inputs["image_encoder_output"]["vae_encoder_out"] = None
         latents, generator = self.run()
         self.end_run()