ランキングに戻る

bytedance/LatentSync

Pythonarxiv.org/abs/2412.09262

Taming Stable Diffusion for Lip Sync!

researchdiffusion-modelslipsyncvideo-genvirtual-avatars
スター成長
スター
5.9k
フォーク
962
週間成長
Issue
212
2k4k
2024年12月2025年6月2026年1月2026年7月
成果物PyPIpip install latentsync
README

LatentSync

arXiv arXiv arXiv Replicate

🔥 Updates

  • 2025/06/11: We released LatentSync 1.6, which is trained on 512 $\times$ 512 resolution videos to mitigate the blurriness problem. Watch the demo here.

  • 2025/03/14: We released LatentSync 1.5, which (1) improves temporal consistency via adding temporal layer, (2) improves performance on Chinese videos and (3) reduces the VRAM requirement of the stage2 training to 20 GB through a series of optimizations. Learn more details here.

📖 Introduction

We present LatentSync, an end-to-end lip-sync method based on audio-conditioned latent diffusion models without any intermediate motion representation, diverging from previous diffusion-based lip-sync methods based on pixel-space diffusion or two-stage generation. Our framework can leverage the powerful capabilities of Stable Diffusion to directly model complex audio-visual correlations.

🏗️ Framework

LatentSync uses the Whisper to convert melspectrogram into audio embeddings, which are then integrated into the U-Net via cross-attention layers. The reference and masked frames are channel-wise concatenated with noised latents as the input of U-Net. In the training process, we use a one-step method to get estimated clean latents from predicted noises, which are then decoded to obtain the estimated clean frames. The TREPA, LPIPS and SyncNet losses are added in the pixel space.

🎬 Demo

Original video Lip-synced video

(Photorealistic videos are filmed by contracted models, and anime videos are from VASA-1)

📑 Open-source Plan

  • Inference code and checkpoints
  • Data processing pipeline
  • Training code

🔧 Setting up the Environment

Install the required packages and download the checkpoints via:

source setup_env.sh

If the download is successful, the checkpoints should appear as follows:

./checkpoints/
|-- latentsync_unet.pt
|-- whisper
|   `-- tiny.pt

Or you can download latentsync_unet.pt and tiny.pt manually from our HuggingFace repo

🚀 Inference

Minimum VRAM for inference:

  • 8 GB with LatentSync 1.5
  • 18 GB with LatentSync 1.6

There are two ways to perform inference:

1. Gradio App

Run the Gradio app for inference:

python gradio_app.py

2. Command Line Interface

Run the script for inference:

./inference.sh

You can try adjusting the following inference parameters to achieve better results:

  • inference_steps [20-50]: A higher value improves visual quality but slows down the generation speed.
  • guidance_scale [1.0-3.0]: A higher value improves lip-sync accuracy but may cause the video distortion or jitter.

🔄 Data Processing Pipeline

The complete data processing pipeline includes the following steps:

  1. Remove the broken video files.
  2. Resample the video FPS to 25, and resample the audio to 16000 Hz.
  3. Scene detect via PySceneDetect.
  4. Split each video into 5-10 second segments.
  5. Affine transform the faces according to the landmarks detected by InsightFace, then resize to 256 $\times$ 256.
  6. Remove videos with sync confidence score lower than 3, and adjust the audio-visual offset to 0.
  7. Calculate hyperIQA score, and remove videos with scores lower than 40.

Run the script to execute the data processing pipeline:

./data_processing_pipeline.sh

You should change the parameter input_dir in the script to specify the data directory to be processed. The processed videos will be saved in the high_visual_quality directory. Each step will generate a new directory to prevent the need to redo the entire pipeline in case the process is interrupted by an unexpected error.

🏋️‍♂️ Training U-Net

Before training, you should process the data as described above. We released a pretrained SyncNet with 94% accuracy on both VoxCeleb2 and HDTF datasets for the supervision of U-Net training. You can execute the following command to download this SyncNet checkpoint:

huggingface-cli download ByteDance/LatentSync-1.6 stable_syncnet.pt --local-dir checkpoints

If all the preparations are complete, you can train the U-Net with the following script:

./train_unet.sh

We prepared several UNet configuration files in the configs/unet directory, each corresponding to a specific training setup:

  • stage1.yaml: Stage1 training, requires 23 GB VRAM.
  • stage2.yaml: Stage2 training with optimal performance, requires 30 GB VRAM.
  • stage2_efficient.yaml: Efficient Stage 2 training, requires 20 GB VRAM. It may lead to slight degradation in visual quality and temporal consistency compared with stage2.yaml, suitable for users with consumer-grade GPUs, such as the RTX 3090.
  • stage1_512.yaml: Stage1 training on 512 $\times$ 512 resolution videos, requires 30 GB VRAM.
  • stage2_512.yaml: Stage2 training on 512 $\times$ 512 resolution videos, requires 55 GB VRAM.

Also remember to change the parameters in U-Net config file to specify the data directory, checkpoint save path, and other training hyperparameters. For convenience, we prepared a script for writing a data files list. Run the following command:

python -m tools.write_fileslist

🏋️‍♂️ Training SyncNet

In case you want to train SyncNet on your own datasets, you can run the following script. The data processing pipeline for SyncNet is the same as U-Net.

./train_syncnet.sh

After validations_steps training, the loss charts will be saved in train_output_dir. They contain both the training and validation loss. If you want to customize the architecture of SyncNet for different image resolutions and input frame lengths, please follow the guide.

📊 Evaluation

You can evaluate the sync confidence score of a generated video by running the following script:

./eval/eval_sync_conf.sh

You can evaluate the accuracy of SyncNet on a dataset by running the following script:

./eval/eval_syncnet_acc.sh

Note that our released SyncNet is trained on data processed through our data processing pipeline, which includes special operations such as affine transformation and audio-visual adjustment. Therefore, before evaluation, the test data must first be processed using the provided pipeline.

🙏 Acknowledgement

Thanks for their generous contributions to the open-source community!

📖 Citation

If you find our repo useful for your research, please consider citing our paper:

@article{li2024latentsync,
  title={LatentSync: Taming Audio-Conditioned Latent Diffusion Models for Lip Sync with SyncNet Supervision},
  author={Li, Chunyu and Zhang, Chao and Xu, Weikai and Lin, Jingyu and Xie, Jinghui and Feng, Weiguo and Peng, Bingyue and Chen, Cunjian and Xing, Weiwei},
  journal={arXiv preprint arXiv:2412.09262},
  year={2024}
}
関連リポジトリ
mvanhorn/last30days-skill

AI agent skill that researches any topic across Reddit, X, YouTube, HN, Polymarket, and the web - then synthesizes a grounded summary

PythonPyPIMIT Licenseai-promptsclaude
53k4.6k
microsoft/qlib

Qlib is an AI-oriented Quant investment platform that aims to use AI tech to empower Quant Research, from exploring ideas to implementing productions. Qlib supports diverse ML modeling paradigms, including supervised learning, market dynamics modeling, and RL, and is now equipped with https://github.com/microsoft/RD-Agent to automate R&D process.

PythonPyPIMIT Licensequantitative-financemachine-learning
qlib.readthedocs.io/en/latest/
46.5k7.4k
google-research/google-research

Google Research

Jupyter NotebookApache License 2.0machine-learningai
research.google
38.4k8.5k
khoj-ai/khoj

Your AI second brain. Self-hostable. Get answers from the web or your docs. Build custom agents, schedule automations, do deep research. Turn any online or local LLM into your personal, autonomous AI (gpt, claude, gemini, llama, qwen, mistral). Get started - free.

PythonPyPIGNU Affero General Public License v3.0semantic-searchemacs
khoj.dev
35.9k2.3k
assafelovic/gpt-researcher

An autonomous agent that conducts deep research on any data using any LLM providers

PythonPyPIApache License 2.0aipython
gptr.dev
28.5k3.9k
dzhng/deep-research

An AI-powered research assistant that performs iterative, deep research on any topic by combining search engines, web scraping, and large language models. The goal of this repo is to provide the simplest implementation of a deep research agent - e.g. an agent that can refine its research direction overtime and deep dive into a topic.

TypeScriptnpmMIT Licenseagentai
19.4k2k
microsoft/AirSim

Open source simulator for autonomous vehicles built on Unreal Engine / Unity, from Microsoft AI & Research

C++Otherdronesai
microsoft.github.io/AirSim/
18.3k4.9k
carla-simulator/carla

Open-source simulator for autonomous driving research.

C++MIT Licensesimulatorautonomous-vehicles
carla.org
14.2k4.6k
microsoft/RD-Agent

Research and development (R&D) is crucial for the enhancement of industrial productivity, especially in the AI era, where the core aspects of R&D are mainly focused on data and models. We are committed to automating these high-value generic R&D processes through R&D-Agent, which lets AI drive data-driven AI. 🔗https://aka.ms/RD-Agent-Tech-Report

PythonPyPIMIT Licenseagentai
rdagent.azurewebsites.net
14k1.8k
bytedance/UI-TARS

Pioneering Automated GUI Interaction with Native Agents

PythonPyPIApache License 2.0research
11.2k852
majd/ipatool

Command-line tool that allows searching and downloading app packages (known as ipa files) from the iOS App Store

GoGo ModulesMIT Licenseiosappstore
9.7k831
LearningCircuit/local-deep-research

~95% on SimpleQA (e.g. Qwen3.6-27B on a 3090). Supports all local and cloud LLMs (llama.cpp, Ollama, Google, ...). 10+ search engines - arXiv, PubMed, your private documents. Everything Local & Encrypted.

PythonPyPIMIT Licenseacademiaarxiv
8.8k769