Skip to content

antgroup/echomimic_v2

Repository files navigation

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.12.16] 🔥 RefImg-Pose Alignment Demo is now available, which involves aligning reference image, extracting pose from driving video, and generating video.
  • [2024.11.27] 🔥 Installation tutorial is now available. Thanks AiMotionStudio for the contribution.
  • [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

ch_intro_git.mp4
en_intro_git.mp4

English Driven Audio

echomimicv2_demo_video.mp4
en_01.mp4
en_02.mp4
en_03.mp4
en_04.mp4
en_05.mp4
en_06.mp4
en_07.mp4
en_08.mp4
en_09.mp4

Chinese Driven Audio

ch_01.mp4
ch_02.mp4
ch_03.mp4
ch_04.mp4
ch_05.mp4
ch_06.mp4
ch_07.mp4
ch_08.mp4
ch_09.mp4

⚒️ Automatic Installation

Download the Codes

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

Automatic Setup

  • CUDA >= 11.7, Python == 3.10
   sh linux_setup.sh

⚒️ Manual 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

Make sure to check out the discussions to learn how to start the inference.

📝 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
Jupyter demo with pose and reference image alignmnet 16st Dec, 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{meng2024echomimicv2,
  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}
}

🌟 Star History

Star History Chart