Update optimization.py

optimization.py

@@ -49,6 +49,13 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
 
     @spaces.GPU(duration=1500)
     def compile_transformer():
+        
+        pipeline.load_lora_weights(
+        "lightx2v/Qwen-Image-Lightning", 
+        weight_name="Qwen-Image-Lightning-8steps-V1.1.safetensors"
+        )
+        pipeline.fuse_lora()
+        pipeline.unload_lora_weights()
 
         with spaces.aoti_capture(pipeline.transformer) as call:
             pipeline(*args, **kwargs)