Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ import numpy as np
|
|
10 |
import PIL.Image
|
11 |
import spaces
|
12 |
import torch
|
13 |
-
from diffusers import AutoencoderKL, DiffusionPipeline
|
14 |
|
15 |
DESCRIPTION = "# SDXL"
|
16 |
if not torch.cuda.is_available():
|
@@ -52,7 +52,7 @@ def generate(
|
|
52 |
use_vae: bool = False,
|
53 |
use_lora: bool = False,
|
54 |
apply_refiner: bool = False,
|
55 |
-
model = '
|
56 |
vaecall = 'stabilityai/sd-vae-ft-mse',
|
57 |
lora = 'amazonaws-la/juliette',
|
58 |
lora_scale: float = 0.7,
|
|
|
10 |
import PIL.Image
|
11 |
import spaces
|
12 |
import torch
|
13 |
+
from diffusers import AutoencoderKL, DiffusionPipeline, DPMSolverMultistepScheduler
|
14 |
|
15 |
DESCRIPTION = "# SDXL"
|
16 |
if not torch.cuda.is_available():
|
|
|
52 |
use_vae: bool = False,
|
53 |
use_lora: bool = False,
|
54 |
apply_refiner: bool = False,
|
55 |
+
model = 'playgroundai/playground-v2.5-1024px-aesthetic',
|
56 |
vaecall = 'stabilityai/sd-vae-ft-mse',
|
57 |
lora = 'amazonaws-la/juliette',
|
58 |
lora_scale: float = 0.7,
|