amazonaws-la commited on
Commit
520d010
·
verified ·
1 Parent(s): 26a7a13

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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, EulerAncestralDiscreteScheduler
14
 
15
  DESCRIPTION = "# SDXL"
16
  if not torch.cuda.is_available():
@@ -61,11 +61,11 @@ def generate(
61
  if torch.cuda.is_available():
62
 
63
  if not use_vae:
64
- scheduler = EulerAncestralDiscreteScheduler.from_pretrained(dropdown_model, subfolder="scheduler")
65
  pipe = DiffusionPipeline.from_pretrained(dropdown_model, scheduler=scheduler, torch_dtype=torch.float16)
66
 
67
  if use_vae:
68
- scheduler = EulerAncestralDiscreteScheduler.from_pretrained(dropdown_model, subfolder="scheduler")
69
  vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16, variant="fp16")
70
  pipe = DiffusionPipeline.from_pretrained(dropdown_model, vae=vae, scheduler=scheduler, torch_dtype=torch.float16)
71
 
 
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():
 
61
  if torch.cuda.is_available():
62
 
63
  if not use_vae:
64
+ scheduler = DPMSolverMultistepScheduler.from_pretrained(dropdown_model, subfolder="scheduler")
65
  pipe = DiffusionPipeline.from_pretrained(dropdown_model, scheduler=scheduler, torch_dtype=torch.float16)
66
 
67
  if use_vae:
68
+ scheduler = DPMSolverMultistepScheduler.from_pretrained(dropdown_model, subfolder="scheduler")
69
  vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16, variant="fp16")
70
  pipe = DiffusionPipeline.from_pretrained(dropdown_model, vae=vae, scheduler=scheduler, torch_dtype=torch.float16)
71