Spaces:
Running
on
Zero
Running
on
Zero
Update optimized.py
Browse files- 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 |
-
|
| 77 |
-
|
| 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(
|
| 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"),
|