LPX55 commited on
Commit
5ac2a35
·
verified ·
1 Parent(s): 1063084

Update optimized.py

Browse files
Files changed (1) hide show
  1. optimized.py +3 -3
optimized.py CHANGED
@@ -73,8 +73,8 @@ if getattr(pipe, "vae", None) is not None:
73
 
74
  pipe.enable_attention_slicing(1)
75
  # B. Enable Memory Optimizations
76
- # pipe.enable_vae_tiling()
77
- # pipe.enable_xformers_memory_efficient_attention()
78
 
79
  # C. Unified Precision Handling
80
  # for comp in [pipe.unet, pipe.vae, pipe.controlnet]:
@@ -110,7 +110,7 @@ iface = gr.Interface(
110
  inputs=[
111
  gr.Textbox(lines=2, placeholder="Enter your prompt here..."),
112
  gr.Slider(1, 3, value=1, label="Scale"),
113
- gr.Slider(6, 30, value=8, label="Steps"),
114
  gr.Image(type="pil", label="Control Image"),
115
  gr.Slider(0, 1, value=0.6, label="ControlNet Scale"),
116
  gr.Slider(1, 20, value=3.5, label="Guidance Scale"),
 
73
 
74
  pipe.enable_attention_slicing(1)
75
  # B. Enable Memory Optimizations
76
+ pipe.enable_vae_tiling()
77
+ pipe.enable_xformers_memory_efficient_attention()
78
 
79
  # C. Unified Precision Handling
80
  # for comp in [pipe.unet, pipe.vae, pipe.controlnet]:
 
110
  inputs=[
111
  gr.Textbox(lines=2, placeholder="Enter your prompt here..."),
112
  gr.Slider(1, 3, value=1, label="Scale"),
113
+ gr.Slider(2, 20, value=8, label="Steps"),
114
  gr.Image(type="pil", label="Control Image"),
115
  gr.Slider(0, 1, value=0.6, label="ControlNet Scale"),
116
  gr.Slider(1, 20, value=3.5, label="Guidance Scale"),