Views
No views yet
1import torch
2from diffusers import DiffusionPipeline
3from diffusers.utils import make_image_grid
4from PIL import Image
5
6pipe = DiffusionPipeline.from_pretrained(
7 "kiigii/imagedream-ipmv-diffusers",
8 torch_dtype=torch.float16,
9 trust_remote_code=True,
10)
11pipe.load_ip_adapter()
12pipe.to("cude")
13
14prompt = "" # no need to input prompt
15image = Image.open(...)
16
17mv_images = pipe(
18 prompt=prompt,
19 ip_adapter_image=image,
20 guidance_scale=5,
21 num_inference_steps=30,
22 elevation=0,
23 num_images_per_prompt=1
24).images
25mv_grid = make_image_grid(mv_images[:4], 2, 2)
26mv_grid.save("mv_image.png")@article{wang2023imagedream,
title={ImageDream: Image-Prompt Multi-view Diffusion for 3D Generation},
author={Wang, Peng and Shi, Yichun},
journal={arXiv preprint arXiv:2312.02201},
year={2023}
}