Update optimization.py

optimization.py

@@ -43,11 +43,11 @@ 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"
-        # )
+        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(
@@ -65,11 +65,11 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
 
 
 
-        pipeline.load_lora_weights(
-            "Kijai/WanVideo_comfy", 
-            weight_name="Pusa/Wan22_PusaV1_lora_HIGH_resized_dynamic_avg_rank_98_bf16.safetensors", 
-            adapter_name="lightning"
-        )
+        # pipeline.load_lora_weights(
+        #     "Kijai/WanVideo_comfy", 
+        #     weight_name="Pusa/Wan22_PusaV1_lora_HIGH_resized_dynamic_avg_rank_98_bf16.safetensors", 
+        #     adapter_name="lightning"
+        # )
         kwargs_lora = {}
         kwargs_lora["load_into_transformer_2"] = True
         # pipeline.load_lora_weights(