Spaces:
Running
on
Zero
Running
on
Zero
alfredplpl
commited on
Commit
·
b1f69aa
1
Parent(s):
4217bef
Update app.py
Browse files
app.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
from diffusers import StableDiffusionXLPipeline, StableDiffusionXLImg2ImgPipeline, EulerAncestralDiscreteScheduler
|
2 |
-
from diffusers import
|
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 =
|
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,
|