amazonaws-sp commited on
Commit
e5bc8f3
·
verified ·
1 Parent(s): fcf1f5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -7
app.py CHANGED
@@ -57,13 +57,6 @@ def generate(
57
  ):
58
  if torch.cuda.is_available():
59
 
60
- if not use_vae:
61
- pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
62
-
63
- if use_vae:
64
- vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)
65
- pipe = DiffusionPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16)
66
-
67
  if use_img2img:
68
  pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model, torch_dtype=torch.float16)
69
 
 
57
  ):
58
  if torch.cuda.is_available():
59
 
 
 
 
 
 
 
 
60
  if use_img2img:
61
  pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model, torch_dtype=torch.float16)
62