amazonaws-la commited on
Commit
b2aa6a1
·
verified ·
1 Parent(s): 56ad88b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = 'SG161222/Realistic_Vision_V6.0_B1_noVAE',
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,