Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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-
|
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',
|