EchoMimicV2 / README.md
lymhust's picture
Update README.md
4b0c02e verified

EchoMimicV2: Towards Striking, Simplified, and Semi-Body Human Animation

Terminal Technology Department, Alipay, Ant Group.

๐Ÿš€ EchoMimic Series

  • EchoMimicV1: Lifelike Audio-Driven Portrait Animations through Editable Landmark Conditioning. GitHub
  • EchoMimicV2: Towards Striking, Simplified, and Semi-Body Human Animation. GitHub

๐Ÿ“ฃ Updates

  • [2024.11.27] ๐Ÿ”ฅ Thanks AiMotionStudio for the installation tutorial.
  • [2024.11.22] ๐Ÿ”ฅ GradioUI is now available. Thanks @gluttony-10 for the contribution.
  • [2024.11.22] ๐Ÿ”ฅ ComfyUI is now available. Thanks @smthemex for the contribution.
  • [2024.11.21] ๐Ÿ”ฅ We release the EMTD dataset list and processing scripts.
  • [2024.11.21] ๐Ÿ”ฅ We release our EchoMimicV2 codes and models.
  • [2024.11.15] ๐Ÿ”ฅ Our paper is in public on arxiv.

๐ŸŒ… Gallery

Introduction

English Driven Audio

Chinese Driven Audio

โš’๏ธ Installation

Download the Codes

  git clone https://github.com/antgroup/echomimic_v2
  cd echomimic_v2

Python Environment Setup

  • Tested System Environment: Centos 7.2/Ubuntu 22.04, Cuda >= 11.7
  • Tested GPUs: A100(80G) / RTX4090D (24G) / V100(16G)
  • Tested Python Version: 3.8 / 3.10 / 3.11

Create conda environment (Recommended):

  conda create -n echomimic python=3.10
  conda activate echomimic

Install packages with pip

  pip install pip -U
  pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 xformers==0.0.28.post3 --index-url https://download.pytorch.org/whl/cu124
  pip install torchao --index-url https://download.pytorch.org/whl/nightly/cu124
  pip install -r requirements.txt
  pip install --no-deps facenet_pytorch==2.6.0

Download ffmpeg-static

Download and decompress ffmpeg-static, then

export FFMPEG_PATH=/path/to/ffmpeg-4.4-amd64-static

Download pretrained weights

git lfs install
git clone https://huggingface.co/BadToBest/EchoMimicV2 pretrained_weights

The pretrained_weights is organized as follows.

./pretrained_weights/
โ”œโ”€โ”€ denoising_unet.pth
โ”œโ”€โ”€ reference_unet.pth
โ”œโ”€โ”€ motion_module.pth
โ”œโ”€โ”€ pose_encoder.pth
โ”œโ”€โ”€ sd-vae-ft-mse
โ”‚   โ””โ”€โ”€ ...
โ”œโ”€โ”€ sd-image-variations-diffusers
โ”‚   โ””โ”€โ”€ ...
โ””โ”€โ”€ audio_processor
    โ””โ”€โ”€ tiny.pt

In which denoising_unet.pth / reference_unet.pth / motion_module.pth / pose_encoder.pth are the main checkpoints of EchoMimic. Other models in this hub can be also downloaded from it's original hub, thanks to their brilliant works:

Inference on Demo

Run the gradio:

python app.py

Run the python inference script:

python infer.py --config='./configs/prompts/infer.yaml'

EMTD Dataset

Download dataset:

python ./EMTD_dataset/download.py

Slice dataset:

bash ./EMTD_dataset/slice.sh

Process dataset:

python ./EMTD_dataset/preprocess.py

๐Ÿ“ Release Plans

Status Milestone ETA
โœ… The inference source code of EchoMimicV2 meet everyone on GitHub 21st Nov, 2024
โœ… Pretrained models trained on English and Mandarin Chinese on HuggingFace 21st Nov, 2024
โœ… Pretrained models trained on English and Mandarin Chinese on ModelScope 21st Nov, 2024
โœ… EMTD dataset list and processing scripts 21st Nov, 2024
๐Ÿš€ Accelerated models to be released TBD
๐Ÿš€ Online Demo on ModelScope to be released TBD
๐Ÿš€ Online Demo on HuggingFace to be released TBD

โš–๏ธ Disclaimer

This project is intended for academic research, and we explicitly disclaim any responsibility for user-generated content. Users are solely liable for their actions while using the generative model. The project contributors have no legal affiliation with, nor accountability for, users' behaviors. It is imperative to use the generative model responsibly, adhering to both ethical and legal standards.

๐Ÿ™๐Ÿป Acknowledgements

We would like to thank the contributors to the MimicMotion and Moore-AnimateAnyone repositories, for their open research and exploration.

We are also grateful to CyberHost and Vlogger for their outstanding work in the area of audio-driven human animation.

If we missed any open-source projects or related articles, we would like to complement the acknowledgement of this specific work immediately.

๐Ÿ“’ Citation

If you find our work useful for your research, please consider citing the paper :

@misc{meng2024echomimic,
  title={EchoMimicV2: Towards Striking, Simplified, and Semi-Body Human Animation},
  author={Rang Meng, Xingyu Zhang, Yuming Li, Chenguang Ma},
  year={2024},
  eprint={2411.10061},
  archivePrefix={arXiv},
  primaryClass={cs.CV}
}

๐ŸŒŸ Star History

Star History Chart