abreza commited on
Commit
ee60c56
1 Parent(s): 3f01a25

fix zero123 path issue

Browse files
Files changed (1) hide show
  1. launch/model_generation.py +1 -1
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(