chaoxu commited on
Commit
5634871
1 Parent(s): ba5ab8d

Zero123++ v1.2

Browse files
Files changed (1) hide show
  1. gradio_app.py +2 -2
gradio_app.py CHANGED
@@ -150,8 +150,8 @@ def gen_multiview(pipeline, predictor, input_image, scale_slider, steps_slider,
150
  def run_demo():
151
  # Load the pipeline
152
  pipeline = DiffusionPipeline.from_pretrained(
153
- "sudo-ai/zero123plus-v1.1", custom_pipeline="sudo-ai/zero123plus-pipeline",
154
- torch_dtype=torch.float16
155
  )
156
  # Feel free to tune the scheduler
157
  pipeline.scheduler = EulerAncestralDiscreteScheduler.from_config(
 
150
  def run_demo():
151
  # Load the pipeline
152
  pipeline = DiffusionPipeline.from_pretrained(
153
+ "sudo-ai/zero123plus-v1.2", custom_pipeline="sudo-ai/zero123plus-pipeline",
154
+ torch_dtype=torch.float16, use_auth_token=os.environ["HF_TOKEN"]
155
  )
156
  # Feel free to tune the scheduler
157
  pipeline.scheduler = EulerAncestralDiscreteScheduler.from_config(