Noename commited on
Commit
765fbda
·
verified ·
1 Parent(s): cab137e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -81,7 +81,7 @@ else:
81
  # CPU-specific optimizations
82
  pipe = pipe.to(device)
83
  # pipe.enable_sequential_cpu_offload()
84
- # pipe.enable_attention_slicing()
85
 
86
  pipe.safety_checker = None
87
  pipe.requires_safety_checker = False
@@ -191,7 +191,7 @@ with block:
191
  strength = gr.Slider(label="Control Strength", minimum=0.0, maximum=2.0, value=1.0, step=0.01)
192
  # guess_mode = gr.Checkbox(label='Guess Mode', value=False)
193
  ddim_steps = gr.Slider(label="Steps", minimum=1, maximum=20, value=20, step=1)
194
- scale = gr.Slider(label="Guidance Scale", minimum=0.1, maximum=30.0, value=7.0, step=0.1)
195
  threshold = gr.Slider(label="Segmentation Threshold", minimum=0.1, maximum=0.9, value=0.5, step=0.05)
196
  seed = gr.Slider(label="Seed", minimum=-1, maximum=2147483647, value=-1, step=1)
197
  eta = gr.Number(label="eta (DDIM)", value=0.0)
 
81
  # CPU-specific optimizations
82
  pipe = pipe.to(device)
83
  # pipe.enable_sequential_cpu_offload()
84
+ pipe.enable_attention_slicing()
85
 
86
  pipe.safety_checker = None
87
  pipe.requires_safety_checker = False
 
191
  strength = gr.Slider(label="Control Strength", minimum=0.0, maximum=2.0, value=1.0, step=0.01)
192
  # guess_mode = gr.Checkbox(label='Guess Mode', value=False)
193
  ddim_steps = gr.Slider(label="Steps", minimum=1, maximum=20, value=20, step=1)
194
+ scale = gr.Slider(label="Guidance Scale", minimum=0.1, maximum=30.0, value=4.0, step=0.1)
195
  threshold = gr.Slider(label="Segmentation Threshold", minimum=0.1, maximum=0.9, value=0.5, step=0.05)
196
  seed = gr.Slider(label="Seed", minimum=-1, maximum=2147483647, value=-1, step=1)
197
  eta = gr.Number(label="eta (DDIM)", value=0.0)