prithivMLmods commited on
Commit
3452ed7
·
verified ·
1 Parent(s): 75a189b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -114,7 +114,7 @@ pipe.load_lora_weights("dx8152/Qwen-Image-Edit-2509-Relight",
114
 
115
  pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
116
  # It's recommended to run optimization after loading all weights
117
- # optimize_pipeline_(pipe, image=[Image.new("RGB", (1024, 1024)), Image.new("RGB", (1024, 1024))], prompt="prompt")
118
 
119
  MAX_SEED = np.iinfo(np.int32).max
120
 
@@ -241,8 +241,8 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
241
  guidance_scale = gr.Slider(label="Guidance Scale", minimum=1.0, maximum=10.0, step=0.1, value=1.0)
242
  steps = gr.Slider(label="Inference Steps", minimum=1, maximum=50, step=1, value=4)
243
  # Hidden sliders to hold image dimensions
244
- height = gr.Slider(label="Height", minimum=256, maximum=1024, step=8, value=1024, visible=False)
245
- width = gr.Slider(label="Width", minimum=256, maximum=1024, step=8, value=1024, visible=False)
246
 
247
  gr.Examples(
248
  examples=[
 
114
 
115
  pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
116
  # It's recommended to run optimization after loading all weights
117
+ optimize_pipeline_(pipe, image=[Image.new("RGB", (1024, 1024)), Image.new("RGB", (1024, 1024))], prompt="prompt")
118
 
119
  MAX_SEED = np.iinfo(np.int32).max
120
 
 
241
  guidance_scale = gr.Slider(label="Guidance Scale", minimum=1.0, maximum=10.0, step=0.1, value=1.0)
242
  steps = gr.Slider(label="Inference Steps", minimum=1, maximum=50, step=1, value=4)
243
  # Hidden sliders to hold image dimensions
244
+ height = gr.Slider(label="Height", minimum=256, maximum=4096, step=8, value=1024, visible=True)
245
+ width = gr.Slider(label="Width", minimum=256, maximum=4096, step=8, value=1024, visible=True)
246
 
247
  gr.Examples(
248
  examples=[