Instructions to use TaoLiveAIGC/AptAvatar with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use TaoLiveAIGC/AptAvatar with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("TaoLiveAIGC/AptAvatar", torch_dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
- AptAvatar: Fast and Vivid Long-Form Audio-Driven Video Generation for Production-Ready Avatars
AptAvatar: Fast and Vivid Long-Form Audio-Driven Video Generation for Production-Ready Avatars
Links are placeholders for now and will be updated as project pages, model cards, community channels, and reports become available.
โจ Key Features
- Endpoint-Anchored Distribution Distillation: It augments vanilla distribution matching with a dedicated Anchor Score Estimator trained on the trajectory-endpoint distribution defined from a frozen pretrained 4-step bridge generator. This provides an attainable endpoint-level anchor for the evolving two-step student.
- Self-Generated History Replay: It reuses cached outputs from earlier generator checkpoints as history conditions during chunk-wise training. This approximates inference-time self-conditioning, mitigating quality degradation from accumulated history errors
๐ฅ News
- 2026.06.29 - Released the initial AptAvatar repository with open-source inference code.
๐ Todo
- Technical report
- Project page
- Inference code
- Checkpoint release
- Additional demos
๐งฉ Pipeline
๐ฌ Examples
๐ Quick Start
๐ ๏ธ Installation
Create and activate the environment:
conda create -n aptavatar python=3.10
conda activate aptavatar
Install PyTorch:
pip install torch==2.8.0 torchvision==0.23.0 --index-url https://download.pytorch.org/whl/cu128
Install project dependencies:
pip install -r requirements.txt
Install FlashAttention:
pip install ninja
pip install flash_attn==2.8.0.post2 --no-build-isolation
If
flash_attn_3is installed in your environment, please uninstall it, as it conflicts with the FlashAttention-3 kernels bundled inxformers:pip uninstall flash_attn_3
Install FFmpeg:
# Ubuntu / Debian
apt-get install ffmpeg
# CentOS / RHEL
yum install ffmpeg ffmpeg-devel
or install through conda:
conda install -c conda-forge ffmpeg==7
๐ฆ Model Preparation
The current repository provides inference code only. After checkpoints are released, place them under the following paths:
| Component | Expected Path | Status |
|---|---|---|
| AptAvatar checkpoint | models/AptAvatar |
AptAvatar |
| Audio encoder | models/chinese-wav2vec2-base |
chinese-wav2vec2-base |
# Placeholder commands. Replace the repository names after release.
pip install "huggingface_hub[cli]"
huggingface-cli download TaoLiveAIGC/AptAvatar --local-dir ./models/AptAvatar
huggingface-cli download TencentGameMate/chinese-wav2vec2-base --local-dir ./models/chinese-wav2vec2-base
๐ฅ Inference
Single-GPU inference:
bash inference_script_single_gpu.sh
Multi-GPU inference:
bash inference_script_multi_gpu.sh
The default scripts use:
examples/woman.png
examples/woman.wav
examples/woman.txt
You can edit the shell scripts or call generate_video.py directly with your own image, audio, and prompt.
๐ Structured Prompt
Besides free-form prompts, AptAvatar supports structured prompts that describe the motion step by step with explicit frame ranges. See examples/woman_template.txt for full examples. Each prompt follows the format:
ๆญฅ้ชค1๏ผ*ๅธง <start>~<end>*<motion description>ๆญฅ้ชค2๏ผ*ๅธง <start>~<end>*<motion description>ๆญฅ้ชค3๏ผ*ๅธง <start>~<end>*<motion description>
- Each step starts with
ๆญฅ้ชคx๏ผfollowed by a frame range marker*ๅธง xx~xx*and a motion description. - We recommend using 1~3 steps per prompt.
Example:
ๆญฅ้ชค1๏ผ*ๅธง 0~20*ๅฅณไธปๆญ้ขๅ้ๅคด็ซ็ซ๏ผๅณๆๅ
ๆฌ่ตท่ณ่ธๅ้ซๅบฆๅนถๅผ ๅผๆๆ๏ผๆๅฟๆๅ๏ผ่บซไฝไฟๆ็จณๅฎ๏ผ่ง็บฟ้ขๅ้ๅคดใๆญฅ้ชค2๏ผ*ๅธง 25~45*ๅฅน็ๅณๆ็ฑๅผ ๅผ็ถๆ้ๆธๆถๆขๆๆพๆณ๏ผๆ่
ๅๆๆๅจไฝๆธ
ๆฐ๏ผๅคด้จๅ่บซไฝไฟๆๆญฃๅ็จณๅฎใๆญฅ้ชค3๏ผ*ๅธง 50~70*ๅฅณไธปๆญๅทฆๆ้ๅๆฌ่ตท่ณไธๅณ่ฉ็ธ่ฟ็้ซๅบฆ๏ผๅๆๅๆถๅผ ๅผๅนถๅ่บซไฝไธคไพงๅฑๅผ๏ผๅฝขๆๅฏน็งฐ็ๅฑ็คบๅงฟๆๅไฟๆๅฎๆ ผใ
๐ Evaluation Results
We evaluate AptAvatar on our internal short-form benchmark for audio-driven avatar generation. Results will be reported at 720p.
Metrics
| Category | Metrics | Direction |
|---|---|---|
| Audio-Visual Alignment | Sync-C, Sync-D | Sync-C higher is better; Sync-D lower is better |
| Image Quality | FID | Lower is better |
| Aesthetic Analysis | ASE, Q-Align | Higher is better |
| VBench | Subject Consistency (Subject-C), Background Consistency (BG-C), Motion Smoothness (Motion-S), and Temporal | |
| Flickering (Temporal-F) | higher is better |
Compared Methods
| Group | Method | NFE |
|---|---|---|
| Many-NFE | InfiniteTalk | 120 |
| Many-NFE | Wan-S2V | 80 |
| Few-NFE | LongCat-Video-Avatar 1.5 | 8 |
| Few-NFE | LiveAvatar | 4 |
| Few-NFE | SoulX-FlashTalk | 4 |
| Few-NFE | AptAvatar | 2 |
720p Results on our internal short-form benchmark
| Method | Group | NFE | Sync-C โ | Sync-D โ | FID โ | ASE โ | Subject-C โ | BG-C โ | Motion Smoothness โ | Temporal Flickering โ |
|---|---|---|---|---|---|---|---|---|---|---|
| InfiniteTalk | Many-NFE | 120 | 7.172 | 8.010 | 24.808 | 3.146 | 96.889 | 95.568 | 99.512 | 99.318 |
| Wan-S2V | Many-NFE | 80 | 5.626 | 9.376 | 43.542 | 3.119 | 97.151 | 95.663 | 99.385 | 99.216 |
| LongCat-Video-Avatar 1.5 | Few-NFE | 8 | 4.140 | 11.433 | 36.582 | 3.191 | 96.338 | 94.533 | 99.550 | 99.332 |
| LiveAvatar | Few-NFE | 4 | 5.986 | 8.586 | 37.868 | 3.184 | 97.804 | 95.718 | 99.409 | 99.226 |
| SoulX-FlashTalk | Few-NFE | 4 | 7.743 | 7.912 | 28.039 | 3.172 | 98.580 | 96.530 | 99.550 | 99.434 |
| AptAvatar | Few-NFE | 2 | 7.865 | 7.739 | 24.759 | 3.307 | 98.589 | 96.844 | 99.561 | 99.456 |
๐ Repository Layout
AptAvatar/
โโโ README.md
โโโ LICENSE
โโโ CHANGELOG.md
โโโ requirements.txt
โโโ config.json
โโโ generate_video.py
โโโ gradio_app.py
โโโ inference_script_single_gpu.sh
โโโ inference_script_multi_gpu.sh
โโโ examples/
โ โโโ woman.png
โ โโโ woman.wav
โ โโโ woman.txt
โโโ assets/
โ โโโ *.mp4
โโโ sample_results/
โ โโโ *.mp4
โโโ models/ # created after downloading checkpoints
โ โโโ AptAvatar/
โ โ โโโ config.json
โ โ โโโ configuration.json
โ โ โโโ diffusion_pytorch_model-0000*-of-00008.safetensors
โ โ โโโ diffusion_pytorch_model.safetensors.index.json
โ โ โโโ Wan2.1_VAE.pth
โ โ โโโ models_clip_open-clip-xlm-roberta-large-vit-huge-14.pth
โ โ โโโ models_t5_umt5-xxl-enc-bf16.pth
โ โ โโโ google/
โ โ โ โโโ umt5-xxl/ # T5 tokenizer files
โ โ โโโ xlm-roberta-large/ # CLIP tokenizer files
โ โโโ chinese-wav2vec2-base/
โ โโโ config.json
โ โโโ preprocessor_config.json
โ โโโ pytorch_model.bin
โโโ AptAvatar/
โโโ inference.py
โโโ configs/
โโโ infinite_talk/
โโโ src/
โโโ wan/
โ๏ธ License Agreement
This repository is released under the Apache License 2.0. See LICENSE for the full license text.
The current release contains inference code. Model weights, datasets, benchmark assets, demo media, and third-party dependencies may be released under separate terms when they become available. Users are responsible for checking and complying with the licenses of all external models, datasets, audio encoders, and media assets used with this repository.
โ ๏ธ Usage Considerations
AptAvatar is intended for research and responsible application development in audio-driven avatar video generation. Users should carefully evaluate outputs before deployment, especially in identity-sensitive, public-facing, or commercial scenarios.
- Obtain appropriate rights and consent for all input images, audio, prompts, and generated likenesses.
- Do not use the system to impersonate real people, mislead audiences, or generate deceptive content.
- Clearly disclose synthetic or AI-generated avatar content when required by law, platform policy, or product context.
- Test robustness across languages, accents, speaking styles, lighting conditions, identities, and long-form generation settings before production use.
- Follow applicable laws and regulations related to privacy, biometric data, publicity rights, copyright, and content safety.
๐ Citation
Citation information will be added after the technical report is released.
@misc{taoavatar2026,
title = {AptAvatar: Fast and Vivid Long-Form Audio-Driven Video Generation for Production-Ready Avatars},
author = {TBD},
year = {2026},
archivePrefix = {arXiv},
eprint = {TBD},
primaryClass = {cs.CV},
url = {TBD}
}
๐ Acknowledgements
AptAvatar builds on the progress of the open-source video generation and audio-driven avatar communities. We sincerely thank the authors and contributors of the following projects:
- InfiniteTalk and Wan: foundational video generation and audio-conditioned avatar backbones that AptAvatar builds upon.
- Self-Forcing: an important reference codebase for long-horizon self-conditioned video generation.
- DMD and Self-Forcing++: key references for distribution matching distillation and self-forcing-style training.
If you find AptAvatar useful, please also consider starring the original repositories of these foundational methods.
- Downloads last month
- -