Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -11,8 +11,9 @@ import PIL.Image
|
|
11 |
import spaces
|
12 |
import torch
|
13 |
from diffusers import AutoencoderKL, DiffusionPipeline
|
|
|
14 |
|
15 |
-
DESCRIPTION = "# RussianVibe XL 1.0"
|
16 |
if not torch.cuda.is_available():
|
17 |
DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
|
18 |
|
@@ -34,6 +35,7 @@ if torch.cuda.is_available():
|
|
34 |
variant="fp16",
|
35 |
)
|
36 |
pipe.load_lora_weights("0x7o/RussianVibe-XL-v1.0")
|
|
|
37 |
if ENABLE_REFINER:
|
38 |
refiner = DiffusionPipeline.from_pretrained(
|
39 |
"stabilityai/stable-diffusion-xl-refiner-1.0",
|
|
|
11 |
import spaces
|
12 |
import torch
|
13 |
from diffusers import AutoencoderKL, DiffusionPipeline
|
14 |
+
from diffusers import DPMSolverMultistepScheduler
|
15 |
|
16 |
+
DESCRIPTION = "# 🪆 RussianVibe XL 1.0"
|
17 |
if not torch.cuda.is_available():
|
18 |
DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
|
19 |
|
|
|
35 |
variant="fp16",
|
36 |
)
|
37 |
pipe.load_lora_weights("0x7o/RussianVibe-XL-v1.0")
|
38 |
+
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
39 |
if ENABLE_REFINER:
|
40 |
refiner = DiffusionPipeline.from_pretrained(
|
41 |
"stabilityai/stable-diffusion-xl-refiner-1.0",
|