Refactored Loras

2025-03-14 23:43:04 +01:00
parent 48b777ae6c
commit d233dd7ed9
7 changed files with 344 additions and 115 deletions
--- a/wan/image2video.py
+++ b/wan/image2video.py
@@ -331,7 +331,7 @@ class WanI2V:
            callback(-1, None)

        for i, t in enumerate(tqdm(timesteps)):
-            offload.set_step_no_for_lora(i)
+            offload.set_step_no_for_lora(self.model, i)
            latent_model_input = [latent.to(self.device)]
            timestep = [t]