Stable-X commited on
Commit
d7bb495
1 Parent(s): 84b9526

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -410,7 +410,7 @@ def main():
410
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
411
 
412
  x_start_pipeline = YOSONormalsPipeline.from_pretrained(
413
- 'Stable-X/yoso-normal-v0-2', trust_remote_code=True, variant="fp16", torch_dtype=torch.float16).to(device)
414
  pipe = StableNormalPipeline.from_pretrained('Stable-X/stable-normal-v0-1', trust_remote_code=True,
415
  variant="fp16", torch_dtype=torch.float16,
416
  scheduler=HEURI_DDIMScheduler(prediction_type='sample',
 
410
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
411
 
412
  x_start_pipeline = YOSONormalsPipeline.from_pretrained(
413
+ 'Stable-X/yoso-normal-v0-3', trust_remote_code=True, variant="fp16", torch_dtype=torch.float16).to(device)
414
  pipe = StableNormalPipeline.from_pretrained('Stable-X/stable-normal-v0-1', trust_remote_code=True,
415
  variant="fp16", torch_dtype=torch.float16,
416
  scheduler=HEURI_DDIMScheduler(prediction_type='sample',