Spaces:

abreza
/

3d_animation_toolkit

Running on Zero

abreza commited on May 10

Commit

ee60c56

•

1 Parent(s): 3f01a25

fix zero123 path issue

Files changed (1) hide show

launch/model_generation.py CHANGED Viewed

@@ -34,7 +34,7 @@ device = torch.device('cuda')
 print('Loading diffusion model ...')
 pipeline = DiffusionPipeline.from_pretrained(
     "sudo-ai/zero123plus-v1.2",
-    custom_pipeline="zero123plus",
     torch_dtype=torch.float16,
 )
 pipeline.scheduler = EulerAncestralDiscreteScheduler.from_config(

 print('Loading diffusion model ...')
 pipeline = DiffusionPipeline.from_pretrained(
     "sudo-ai/zero123plus-v1.2",
+    custom_pipeline="./instantMesh/zero123plus",
     torch_dtype=torch.float16,
 )
 pipeline.scheduler = EulerAncestralDiscreteScheduler.from_config(