| # MVDream-diffusers |
|
|
| A **unified** diffusers implementation of [MVDream](https://github.com/bytedance/MVDream) and [ImageDream](https://github.com/bytedance/ImageDream). |
|
|
| We provide converted `fp16` weights on huggingface: |
| * [MVDream](https://huggingface.co/ashawkey/mvdream-sd2.1-diffusers) |
| * [ImageDream](https://huggingface.co/ashawkey/imagedream-ipmv-diffusers) |
|
|
|
|
| ### Install |
| ```bash |
| # dependency |
| pip install -r requirements.txt |
| |
| # xformers is required! please refer to https://github.com/facebookresearch/xformers |
| pip install ninja |
| pip install -v -U git+https://github.com/facebookresearch/xformers.git@main#egg=xformers |
| ``` |
|
|
| ### Usage |
|
|
| ```bash |
| python run_mvdream.py "a cute owl" |
| python run_imagedream.py data/anya_rgba.png |
| ``` |
|
|
| ### Convert weights |
|
|
| MVDream: |
| ```bash |
| # download original ckpt (we only support the SD 2.1 version) |
| mkdir models |
| cd models |
| wget https://huggingface.co/MVDream/MVDream/resolve/main/sd-v2.1-base-4view.pt |
| wget https://raw.githubusercontent.com/bytedance/MVDream/main/mvdream/configs/sd-v2-base.yaml |
| cd .. |
| |
| # convert |
| python convert_mvdream_to_diffusers.py --checkpoint_path models/sd-v2.1-base-4view.pt --dump_path ./weights_mvdream --original_config_file models/sd-v2-base.yaml --half --to_safetensors --test |
| ``` |
|
|
| ImageDream: |
| ```bash |
| # download original ckpt (we only support the pixel-controller version) |
| cd models |
| wget https://huggingface.co/Peng-Wang/ImageDream/resolve/main/sd-v2.1-base-4view-ipmv.pt |
| wget https://raw.githubusercontent.com/bytedance/ImageDream/main/extern/ImageDream/imagedream/configs/sd_v2_base_ipmv.yaml |
| cd .. |
| |
| # convert |
| python convert_mvdream_to_diffusers.py --checkpoint_path models/sd-v2.1-base-4view-ipmv.pt --dump_path ./weights_imagedream --original_config_file models/sd_v2_base_ipmv.yaml --half --to_safetensors --test |
| ``` |
|
|
| ### Acknowledgement |
|
|
| * The original papers: |
| ```bibtex |
| @article{shi2023MVDream, |
| author = {Shi, Yichun and Wang, Peng and Ye, Jianglong and Mai, Long and Li, Kejie and Yang, Xiao}, |
| title = {MVDream: Multi-view Diffusion for 3D Generation}, |
| journal = {arXiv:2308.16512}, |
| year = {2023}, |
| } |
| @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} |
| } |
| ``` |
| * This codebase is modified from [mvdream-hf](https://github.com/KokeCacao/mvdream-hf). |
