Update optimization.py

optimization.py

@@ -50,6 +50,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)