um235 commited on
Commit
ca7d365
·
verified ·
1 Parent(s): 7082238

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -112,7 +112,7 @@ def infer(
112
  if d_bckg:
113
  image=remove(image)
114
 
115
-
116
  return image, seed
117
 
118
 
@@ -160,8 +160,12 @@ with gr.Blocks(css=css) as demo:
160
  step=0.05,
161
  value=1,
162
  )
163
- with gr.Row():
164
- d_bckg=gr.Checkbox(label="Delete Background", value=True)
 
 
 
 
165
  with gr.Row():
166
  prompt = gr.Text(
167
  label="Prompt",
 
112
  if d_bckg:
113
  image=remove(image)
114
 
115
+ #pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config, rescale_betas_zero_snr=True)
116
  return image, seed
117
 
118
 
 
160
  step=0.05,
161
  value=1,
162
  )
163
+ with gr.Row():
164
+ d_bckg=gr.Checkbox(label="Delete Background", value=False)
165
+ ddim_use=gr.Checkbox(label="Enable DDIMScheduler", value=False)
166
+ distill_vae=gr.Checkbox(label="Use tiny VAE with distill model", value=True)
167
+
168
+ # pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config, rescale_betas_zero_snr=True)
169
  with gr.Row():
170
  prompt = gr.Text(
171
  label="Prompt",