piEsposito commited on
Commit
e816e5f
·
1 Parent(s): 4a92080

move to euler-a

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,4 +1,4 @@
1
- from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline, DPMSolverMultistepScheduler
2
  import gradio as gr
3
  import torch
4
  from PIL import Image
@@ -6,7 +6,7 @@ from PIL import Image
6
  model_id = '22h/vintedois-diffusion-v0-1'
7
  prefix = ''
8
 
9
- scheduler = DPMSolverMultistepScheduler.from_pretrained(model_id, subfolder="scheduler")
10
 
11
  pipe = StableDiffusionPipeline.from_pretrained(
12
  model_id,
 
1
+ from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline, EulerAncestralDiscreteScheduler
2
  import gradio as gr
3
  import torch
4
  from PIL import Image
 
6
  model_id = '22h/vintedois-diffusion-v0-1'
7
  prefix = ''
8
 
9
+ scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
10
 
11
  pipe = StableDiffusionPipeline.from_pretrained(
12
  model_id,