dylanebert HF staff commited on
Commit
06bf081
1 Parent(s): 2dd3270

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,7 +8,7 @@ from PIL import Image
8
  # Text-to-Multi-View Diffusion pipeline
9
  text_pipeline = DiffusionPipeline.from_pretrained(
10
  "dylanebert/mvdream",
11
- custom_pipeline="dylanebert/multi_view_diffusion",
12
  torch_dtype=torch.float16,
13
  trust_remote_code=True,
14
  ).to("cuda")
@@ -16,8 +16,8 @@ text_pipeline = DiffusionPipeline.from_pretrained(
16
 
17
  # Image-to-Multi-View Diffusion pipeline
18
  image_pipeline = DiffusionPipeline.from_pretrained(
19
- "dylanebert/imagedream",
20
- custom_pipeline="dylanebert/multi_view_diffusion",
21
  torch_dtype=torch.float16,
22
  trust_remote_code=True,
23
  ).to("cuda")
 
8
  # Text-to-Multi-View Diffusion pipeline
9
  text_pipeline = DiffusionPipeline.from_pretrained(
10
  "dylanebert/mvdream",
11
+ custom_pipeline="dylanebert/multi-view-diffusion",
12
  torch_dtype=torch.float16,
13
  trust_remote_code=True,
14
  ).to("cuda")
 
16
 
17
  # Image-to-Multi-View Diffusion pipeline
18
  image_pipeline = DiffusionPipeline.from_pretrained(
19
+ "dylanebert/multi-view-diffusion",
20
+ custom_pipeline="dylanebert/multi-view-diffusion",
21
  torch_dtype=torch.float16,
22
  trust_remote_code=True,
23
  ).to("cuda")