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(
|