zwl commited on
Commit
44fe15d
1 Parent(s): ff8b9e9
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -14,6 +14,7 @@ scheduler = UniPCMultistepScheduler(
14
  predict_x0=True,
15
  solver_type="bh2",
16
  lower_order_final=True,
 
17
  )
18
 
19
  class Model:
@@ -25,6 +26,7 @@ class Model:
25
  self.pipe_i2i = None
26
 
27
  models = [
 
28
  Model("Stable-Diffusion-v1.4", "CompVis/stable-diffusion-v1-4", "The 1.4 version of official stable-diffusion"),
29
  Model("Waifu", "hakurei/waifu-diffusion", "anime style"),
30
  ]
@@ -270,4 +272,4 @@ with gr.Blocks(css=css) as demo:
270
  ''')
271
 
272
  demo.queue(concurrency_count=1)
273
- demo.launch(debug=False, share=False)
 
14
  predict_x0=True,
15
  solver_type="bh2",
16
  lower_order_final=True,
17
+ disable_corrector=[0],
18
  )
19
 
20
  class Model:
 
26
  self.pipe_i2i = None
27
 
28
  models = [
29
+ Model("Stable-Diffusion-XL", "stabilityai/stable-diffusion-xl-base-1.0", "The XL version of official stable-diffusion"),
30
  Model("Stable-Diffusion-v1.4", "CompVis/stable-diffusion-v1-4", "The 1.4 version of official stable-diffusion"),
31
  Model("Waifu", "hakurei/waifu-diffusion", "anime style"),
32
  ]
 
272
  ''')
273
 
274
  demo.queue(concurrency_count=1)
275
+ demo.launch(debug=False, share=False)