LGM-mini

Running on Zero

dylanebert HF staff commited on Apr 19, 2024

Commit

04605b1

verified ·

1 Parent(s): 1803df3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ os.makedirs(TMP_DIR, exist_ok=True)
 image_pipeline = DiffusionPipeline.from_pretrained(
     "ashawkey/imagedream-ipmv-diffusers",
-    custom_pipeline="dylanebert/multi_view_diffusion",
     torch_dtype=torch.float16,
     trust_remote_code=True,
 ).to("cuda")

 image_pipeline = DiffusionPipeline.from_pretrained(
     "ashawkey/imagedream-ipmv-diffusers",
+    custom_pipeline="dylanebert/multi-view-diffusion",
     torch_dtype=torch.float16,
     trust_remote_code=True,
 ).to("cuda")