v8hid commited on
Commit
b1750c4
·
1 Parent(s): 6305d2a

changing sampler to euler-a

Browse files
Files changed (1) hide show
  1. zoom.py +2 -2
zoom.py CHANGED
@@ -1,5 +1,5 @@
1
  from helpers import *
2
- from diffusers import StableDiffusionInpaintPipeline, DPMSolverMultistepScheduler
3
  from PIL import Image
4
  import gradio as gr
5
  import numpy as np
@@ -39,7 +39,7 @@ def zoom(
39
  model_id,
40
  torch_dtype=torch.float16,
41
  )
42
- pipe.scheduler = DPMSolverMultistepScheduler.from_config(
43
  pipe.scheduler.config)
44
  pipe = pipe.to("cuda")
45
 
 
1
  from helpers import *
2
+ from diffusers import StableDiffusionInpaintPipeline, EulerAncestralDiscreteScheduler
3
  from PIL import Image
4
  import gradio as gr
5
  import numpy as np
 
39
  model_id,
40
  torch_dtype=torch.float16,
41
  )
42
+ pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(
43
  pipe.scheduler.config)
44
  pipe = pipe.to("cuda")
45