MaxMilan1
commited on
Commit
·
b26070f
1
Parent(s):
2c2acce
change path to custom_pipeline
Browse files
app.py
CHANGED
|
@@ -104,7 +104,7 @@ device = torch.device('cuda')
|
|
| 104 |
print('Loading diffusion model ...')
|
| 105 |
pipeline = DiffusionPipeline.from_pretrained(
|
| 106 |
"sudo-ai/zero123plus-v1.2",
|
| 107 |
-
custom_pipeline="zero123plus",
|
| 108 |
torch_dtype=torch.float16,
|
| 109 |
)
|
| 110 |
pipeline.scheduler = EulerAncestralDiscreteScheduler.from_config(
|
|
|
|
| 104 |
print('Loading diffusion model ...')
|
| 105 |
pipeline = DiffusionPipeline.from_pretrained(
|
| 106 |
"sudo-ai/zero123plus-v1.2",
|
| 107 |
+
custom_pipeline="./zero123plus",
|
| 108 |
torch_dtype=torch.float16,
|
| 109 |
)
|
| 110 |
pipeline.scheduler = EulerAncestralDiscreteScheduler.from_config(
|