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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ from PIL import Image
7
 
8
  # Text-to-Multi-View Diffusion pipeline
9
  text_pipeline = DiffusionPipeline.from_pretrained(
10
- "ashawkey/mvdream-sd2.1-diffusers",
11
  custom_pipeline="dylanebert/multi_view_diffusion",
12
  torch_dtype=torch.float16,
13
  trust_remote_code=True,
@@ -16,7 +16,7 @@ text_pipeline = DiffusionPipeline.from_pretrained(
16
 
17
  # Image-to-Multi-View Diffusion pipeline
18
  image_pipeline = DiffusionPipeline.from_pretrained(
19
- "ashawkey/imagedream-ipmv-diffusers",
20
  custom_pipeline="dylanebert/multi_view_diffusion",
21
  torch_dtype=torch.float16,
22
  trust_remote_code=True,
 
7
 
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,
 
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,