amazonaws-sp commited on
Commit
944fca9
·
verified ·
1 Parent(s): 4cf1a27

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -56,7 +56,7 @@ def generate(
56
  if torch.cuda.is_available():
57
 
58
  if not use_vae:
59
- pipe = DiffusionPipeline.from_single_file("https://civitai.com/api/download/models/403131", torch_dtype=torch.float16)
60
 
61
  if use_vae:
62
  vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)
 
56
  if torch.cuda.is_available():
57
 
58
  if not use_vae:
59
+ pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
60
 
61
  if use_vae:
62
  vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)