Spaces:
0x7o
/
Runtime error

0x7o commited on
Commit
a69dd38
·
verified ·
1 Parent(s): 14ae0c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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",