github

guoqincode / AnimateAnyone-unofficial

  • четверг, 14 декабря 2023 г. в 00:00:06
https://github.com/guoqincode/AnimateAnyone-unofficial

Unofficial Implementation of Animate Anyone



Unofficial Implementation of Animate Anyone

Overview

This repository contains an simple and unofficial implementation of Animate Anyone. This project is built upon magic-animate and AnimateDiff.

Note !!!

This project is under continuous development in part-time, there may be bugs in the code, welcome to correct them, I will optimize the code after the pre-trained model is released!

Requirements

Same as magic-animate.

ToDo

  • Release Training Code.
  • DeepSpeed + Accelerator Training
  • Release Inference Code and unofficial pre-trained weights.

Training

First Stage

torchrun --nnodes=1 --nproc_per_node=1 train.py --config configs/training/train_stage_1.yaml

Second Stage

torchrun --nnodes=1 --nproc_per_node=1 train.py --config configs/training/train_stage_2.yaml

Inference

First Stage

Second Stage

Acknowledgements

Special thanks to the original authors of the Animate Anyone project and the contributors to the magic-animate and AnimateDiff repository for their open research and foundational work that inspired this unofficial implementation.