This repository contains an simple and unofficial implementation of Animate Anyone. This project is built upon magic-animate and AnimateDiff.
The first training phase basic test passed, currently in training and testing the second phase.
Training may be slow due to GPU shortage.😢
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!
Same as magic-animate.
- Release Training Code.
- DeepSpeed + Accelerator Training
- Release Inference Code and unofficial pre-trained weights.
torchrun --nnodes=1 --nproc_per_node=1 train.py --config configs/training/train_stage_1.yamltorchrun --nnodes=1 --nproc_per_node=1 train.py --config configs/training/train_stage_2.yamlpython3 -m pipelines.animation_stage_1 --config configs/prompts/animation_stage_1.yamlSpecial 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.