Spaces:
Runtime error
Runtime error
fix zero123 path issue
Browse files
launch/model_generation.py
CHANGED
@@ -34,7 +34,7 @@ device = torch.device('cuda')
|
|
34 |
print('Loading diffusion model ...')
|
35 |
pipeline = DiffusionPipeline.from_pretrained(
|
36 |
"sudo-ai/zero123plus-v1.2",
|
37 |
-
custom_pipeline="zero123plus",
|
38 |
torch_dtype=torch.float16,
|
39 |
)
|
40 |
pipeline.scheduler = EulerAncestralDiscreteScheduler.from_config(
|
|
|
34 |
print('Loading diffusion model ...')
|
35 |
pipeline = DiffusionPipeline.from_pretrained(
|
36 |
"sudo-ai/zero123plus-v1.2",
|
37 |
+
custom_pipeline="./instantMesh/zero123plus",
|
38 |
torch_dtype=torch.float16,
|
39 |
)
|
40 |
pipeline.scheduler = EulerAncestralDiscreteScheduler.from_config(
|