Update optimization.py

optimization.py

@@ -67,7 +67,7 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
 
         pipeline.load_lora_weights(
             "Kijai/WanVideo_comfy", 
-            weight_name="Lightx2v/lightx2v_T2V_14B_cfg_step_distill_v2_lora_rank128_bf16.safetensors", 
+            weight_name="Pusa/Wan22_PusaV1_lora_HIGH_resized_dynamic_avg_rank_98_bf16.safetensors", 
             adapter_name="lightning"
         )
         kwargs_lora = {}
@@ -86,10 +86,10 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
 
         pipeline.load_lora_weights(
             #"drozbay/Wan2.2_A14B_lora_extract", 
-             "lightx2v/Wan2.2-Lightning", 
+             "Kijai/WanVideo_comfy", 
             #weight_name="MTVCrafter/Wan2_1_MTV-Crafter_motion_adapter_bf16.safetensors", 
              #weight_name="Skyreels/Wan2_1_Skyreels-v2-T2V-720P_LoRA_rank_64_fp16.safetensors",
-            weight_name="Wan2.2-T2V-A14B-4steps-lora-rank64-Seko-V1/low_noise_model.safetensors", 
+            weight_name="Pusa/Wan22_PusaV1_lora_LOW_resized_dynamic_avg_rank_98_bf16.safetensors", 
             #weight_name="Wan2.2-T2V-A14B-4steps-lora-rank64-Seko-V1.1/low_noise_model.safetensors", 
             adapter_name="lightning_2", **kwargs_lora
         )