Spaces:

dylanebert
/

multi-view-diffusion

Running on Zero

dylanebert HF staff commited on May 7

Commit

06bf081

•

1 Parent(s): 2dd3270

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ from PIL import Image
 # Text-to-Multi-View Diffusion pipeline
 text_pipeline = DiffusionPipeline.from_pretrained(
     "dylanebert/mvdream",
-    custom_pipeline="dylanebert/multi_view_diffusion",
     torch_dtype=torch.float16,
     trust_remote_code=True,
 ).to("cuda")
@@ -16,8 +16,8 @@ text_pipeline = DiffusionPipeline.from_pretrained(
 # Image-to-Multi-View Diffusion pipeline
 image_pipeline = DiffusionPipeline.from_pretrained(
-    "dylanebert/imagedream",
-    custom_pipeline="dylanebert/multi_view_diffusion",
     torch_dtype=torch.float16,
     trust_remote_code=True,
 ).to("cuda")

 # Text-to-Multi-View Diffusion pipeline
 text_pipeline = DiffusionPipeline.from_pretrained(
     "dylanebert/mvdream",
+    custom_pipeline="dylanebert/multi-view-diffusion",
     torch_dtype=torch.float16,
     trust_remote_code=True,
 ).to("cuda")
 # Image-to-Multi-View Diffusion pipeline
 image_pipeline = DiffusionPipeline.from_pretrained(
+    "dylanebert/multi-view-diffusion",
+    custom_pipeline="dylanebert/multi-view-diffusion",
     torch_dtype=torch.float16,
     trust_remote_code=True,
 ).to("cuda")