JoPmt commited on
Commit
d32976d
1 Parent(s): 358c486

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,4 +1,4 @@
1
- from diffusers import StableDiffusionXLPipeline, EulerDiscreteScheduler
2
  import torch
3
  from transformers import pipeline
4
  import gradio as gr
@@ -20,7 +20,7 @@ InferenceClient=InferenceClient()
20
  apol=[]
21
 
22
  pipe = accelerator.prepare(StableDiffusionXLPipeline.from_pretrained("Lykon/dreamshaper-xl-lightning", torch_dtype=torch.bfloat16, use_safetensors=True, variant=None, safety_checker=False))
23
- pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
24
  pipe.unet.to(memory_format=torch.channels_last)
25
  pipe = accelerator.prepare(pipe.to("cpu"))
26
 
 
1
+ from diffusers import StableDiffusionXLPipeline, EulerDiscreteScheduler, DPMSolverMultistepScheduler
2
  import torch
3
  from transformers import pipeline
4
  import gradio as gr
 
20
  apol=[]
21
 
22
  pipe = accelerator.prepare(StableDiffusionXLPipeline.from_pretrained("Lykon/dreamshaper-xl-lightning", torch_dtype=torch.bfloat16, use_safetensors=True, variant=None, safety_checker=False))
23
+ pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
24
  pipe.unet.to(memory_format=torch.channels_last)
25
  pipe = accelerator.prepare(pipe.to("cpu"))
26