Qwen-Image-Edit-Relight

Running on Zero

linoyts HF Staff commited on 3 days ago

Commit

8cb8007

verified ·

1 Parent(s): 01f9a93

Update optimization.py

Files changed (1) hide show

optimization.py CHANGED Viewed

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

     @spaces.GPU(duration=1500)
     def compile_transformer():
+        pipeline.load_lora_weights(
         "lightx2v/Qwen-Image-Lightning",
         weight_name="Qwen-Image-Lightning-8steps-V1.1.safetensors"
         )