alfredplpl commited on
Commit
b1f69aa
1 Parent(s): 4217bef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -1,5 +1,5 @@
1
  from diffusers import StableDiffusionXLPipeline, StableDiffusionXLImg2ImgPipeline, EulerAncestralDiscreteScheduler
2
- from diffusers import LCMScheduler, AutoPipelineForText2Image
3
  import gradio as gr
4
  import torch
5
  from PIL import Image
@@ -8,7 +8,6 @@ import os
8
  from huggingface_hub import hf_hub_download
9
  import torch
10
  from torch import autocast
11
- from diffusers import StableDiffusionXLPipeline, DPMSolverMultistepScheduler
12
  from safetensors import safe_open
13
  from compel import Compel, ReturnedEmbeddingsType
14
  from safetensors.torch import load_file
@@ -18,7 +17,7 @@ auth_token=os.environ["ACCESS_TOKEN"]
18
  #adapter_id = "latent-consistency/lcm-lora-sdxl"
19
  #adapter_id_2 = "manual.safetensors"
20
 
21
- scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler",use_auth_token=auth_token)
22
 
23
  pipe = StableDiffusionXLPipeline.from_pretrained(
24
  model_id,
 
1
  from diffusers import StableDiffusionXLPipeline, StableDiffusionXLImg2ImgPipeline, EulerAncestralDiscreteScheduler
2
+ from diffusers import UniPCMultistepScheduler
3
  import gradio as gr
4
  import torch
5
  from PIL import Image
 
8
  from huggingface_hub import hf_hub_download
9
  import torch
10
  from torch import autocast
 
11
  from safetensors import safe_open
12
  from compel import Compel, ReturnedEmbeddingsType
13
  from safetensors.torch import load_file
 
17
  #adapter_id = "latent-consistency/lcm-lora-sdxl"
18
  #adapter_id_2 = "manual.safetensors"
19
 
20
+ scheduler = UniPCMultistepScheduler.from_pretrained(model_id, subfolder="scheduler",use_auth_token=auth_token)
21
 
22
  pipe = StableDiffusionXLPipeline.from_pretrained(
23
  model_id,