Stable-X commited on
Commit
0e763f8
1 Parent(s): c15ed95

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -398,7 +398,7 @@ def main():
398
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
399
 
400
  pipe = YOSODiffusePipeline.from_pretrained(
401
- 'Stable-X/yoso-diffuse-v0-2', trust_remote_code=True, variant="fp16",
402
  torch_dtype=torch.float16, t_start=0).to(device)
403
  try:
404
  import xformers
 
398
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
399
 
400
  pipe = YOSODiffusePipeline.from_pretrained(
401
+ 'Stable-X/yoso-delight-v0-4', trust_remote_code=True, variant="fp16",
402
  torch_dtype=torch.float16, t_start=0).to(device)
403
  try:
404
  import xformers