Update optimization.py

optimization.py

@@ -43,23 +43,22 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
     @spaces.GPU(duration=1500)
     def compile_transformer():
         
-        pipe.load_lora_weights(
-       "Kijai/WanVideo_comfy", 
-       weight_name="Wan22-Lightning/Wan2.2-Lightning_T2V-A14B-4steps-lora_HIGH_fp16.safetensors", 
-        adapter_name="lightning"
-            )
-        kwargs = {}
-        kwargs["load_into_transformer_2"] = True
-        pipe.load_lora_weights(
-           "Kijai/WanVideo_comfy", 
-           weight_name="Wan22-Lightning/Wan2.2-Lightning_T2V-A14B-4steps-lora_LOW_fp16.safetensors", 
-            adapter_name="lightning_2", **kwargs
-        )
-        pipe.set_adapters(["lightning", "lightning_2"], adapter_weights=[1., 1.])
-
-        pipeline.fuse_lora(adapter_names=["lightx2v"], lora_scale=1., components=["transformer"])
-        pipeline.fuse_lora(adapter_names=["lightx2v_2"], lora_scale=1., components=["transformer_2"])
-        pipeline.unload_lora_weights()
+        # pipeline.load_lora_weights(
+        #     "Kijai/WanVideo_comfy", 
+        #     weight_name="Lightx2v/lightx2v_I2V_14B_480p_cfg_step_distill_rank128_bf16.safetensors", 
+        #     adapter_name="lightx2v"
+        # )
+        # kwargs_lora = {}
+        # kwargs_lora["load_into_transformer_2"] = True
+        # pipeline.load_lora_weights(
+        #     "Kijai/WanVideo_comfy", 
+        #     weight_name="Wan22-Lightning/Wan2.2-Lightning_T2V-A14B-4steps-lora_LOW_fp16.safetensors", 
+        #     adapter_name="lightx2v_2", **kwargs_lora
+        # )
+        # pipeline.set_adapters(["lightx2v", "lightx2v_2"], adapter_weights=[1., 1.])
+        # pipeline.fuse_lora(adapter_names=["lightx2v"], lora_scale=3., components=["transformer"])
+        # pipeline.fuse_lora(adapter_names=["lightx2v_2"], lora_scale=1., components=["transformer_2"])
+        # pipeline.unload_lora_weights()
         
         with capture_component_call(pipeline, 'transformer') as call:
             pipeline(*args, **kwargs)