Update optimization.py

optimization.py

@@ -43,79 +43,24 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
     @spaces.GPU(duration=1500)
     def compile_transformer():
         
-        # pipeline.load_lora_weights(
-        #     "Kijai/WanVideo_comfy", 
-        #     weight_name="Lightx2v/lightx2v_T2V_14B_cfg_step_distill_v2_lora_rank128_bf16.safetensors", 
-        #     adapter_name="lightning"
-        # )
-        # kwargs_lora = {}
-        # kwargs_lora["load_into_transformer_2"] = True
-        # pipeline.load_lora_weights(
-        #     "Kijai/WanVideo_comfy", 
-        #      weight_name="Lightx2v/lightx2v_T2V_14B_cfg_step_distill_v2_lora_rank128_bf16.safetensors", 
-        #     #weight_name="Wan22-Lightning/Wan2.2-Lightning_T2V-A14B-4steps-lora_LOW_fp16.safetensors", 
-        #     adapter_name="lightning_2", **kwargs_lora
-        # )
-        # pipeline.set_adapters(["lightning", "lightning_2"], adapter_weights=[1., 1.])
-
-
-        # pipeline.fuse_lora(adapter_names=["lightning"], lora_scale=3., components=["transformer"])
-        # pipeline.fuse_lora(adapter_names=["lightning_2"], lora_scale=1., components=["transformer_2"])
-        # pipeline.unload_lora_weights()
-
-
-
         pipeline.load_lora_weights(
             "Kijai/WanVideo_comfy", 
-            #weight_name="LoRAs/Wan22_Lightx2v/Wan_2_2_I2V_A14B_HIGH_lightx2v_4step_lora_v1030_rank_64_bf16.safetensors", 
-            weight_name="Lightx2v/lightx2v_T2V_14B_cfg_step_distill_v2_lora_rank128_bf16.safetensors",
+            weight_name="Lightx2v/lightx2v_T2V_14B_cfg_step_distill_v2_lora_rank128_bf16.safetensors", 
             adapter_name="lightning"
         )
         kwargs_lora = {}
         kwargs_lora["load_into_transformer_2"] = True
-        # pipeline.load_lora_weights(
-        #     #"drozbay/Wan2.2_A14B_lora_extract", 
-        #      "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="Pusa/Wan21_PusaV1_LoRA_14B_rank512_bf16.safetensors", 
-        #     weight_name="Wan22-Lightning/Wan2.2-Lightning_T2V-A14B-4steps-lora_LOW_fp16.safetensors", 
-        #     adapter_name="lightning_2", **kwargs_lora
-        # )
-
-
-
         pipeline.load_lora_weights(
-            #"drozbay/Wan2.2_A14B_lora_extract", 
-             "lightx2v/Wan2.2-Lightning", 
-            #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="Wan2.2-T2V-A14B-4steps-lora-rank64-Seko-V1.1/low_noise_model.safetensors", 
+            "Kijai/WanVideo_comfy", 
+             weight_name="Lightx2v/lightx2v_T2V_14B_cfg_step_distill_v2_lora_rank128_bf16.safetensors", 
+            #weight_name="Wan22-Lightning/Wan2.2-Lightning_T2V-A14B-4steps-lora_LOW_fp16.safetensors", 
             adapter_name="lightning_2", **kwargs_lora
         )
-
-        # pipeline.load_lora_weights(
-        #     #"drozbay/Wan2.2_A14B_lora_extract", 
-        #      "ostris/wan22_i2v_14b_orbit_shot_lora", 
-        #     #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="wan22_14b_i2v_orbit_low_noise.safetensors", 
-        #     #weight_name="Wan2.2-T2V-A14B-4steps-lora-rank64-Seko-V1.1/low_noise_model.safetensors", 
-        #     adapter_name="lightning_200", **kwargs_lora
-        # )
-
-
-
-
-
-
-        pipeline.set_adapters(["lightning", "lightning_2"], adapter_weights=[1.5, 1.])
+        pipeline.set_adapters(["lightning", "lightning_2"], adapter_weights=[1., 1.])
 
 
         pipeline.fuse_lora(adapter_names=["lightning"], lora_scale=3., components=["transformer"])
         pipeline.fuse_lora(adapter_names=["lightning_2"], lora_scale=1., components=["transformer_2"])
-       
         pipeline.unload_lora_weights()
         
         with capture_component_call(pipeline, 'transformer') as call: