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

Project Page Technical Report Hugging Face License

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_3 is installed in your environment, please uninstall it, as it conflicts with the FlashAttention-3 kernels bundled in xformers:

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
-
Safetensors
Model size
19B params
Tensor type
BF16
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support