Skip to content
 
 

Repository files navigation

JoyAI-Video-Edit

Real-Time Open-Ended Video Editing with Autoregressive Diffusion

Paper Project Hugging Face Demo License

JoyAI-Video-Edit teaser

JoyAI-Video-Edit is a real-time, instruction-guided video editing system for open-ended video streams. Given a live camera stream or uploaded video and a natural-language edit instruction, it edits frames causally as they arrive, without waiting for the full video, requiring a predefined video length, or revisiting future frames. In our deployment benchmark, the full end-to-end pipeline reaches 30.19 FPS at 720x1280, pushing video editing from offline batch processing toward interactive streaming generation.

The system combines an MLLM-based condition encoder, a causal video VAE, and a 16B-parameter multimodal diffusion transformer. It is trained and deployed as an autoregressive diffusion editor, then accelerated with aligned autoregressive distribution matching distillation, long-horizon optimization, bounded KV-state inference, and deployment-oriented scheduling to sustain high-throughput 720p editing while reducing train-inference mismatch and accumulated temporal drift.

🔥🔥🔥 News!!

  • 2026.08.06: 🎉 We are excited to release the live demo of our real-time video editing model—try it here: Demo.
  • 2026.08.05: 🎉 We release the deployment code, technical report, online demo, and JoyAI-Video-Edit checkpoints. Please check the links above for details.

💎 Highlights

  • Real-time open-ended editing. Edits live or uploaded videos as frames arrive, without requiring the full sequence upfront.
  • Diverse instruction control. Supports subject edits, local edits, background changes, style transfer, motion changes, and reference-guided editing.
  • Autoregressive diffusion design. Combines an MLLM condition encoder, causal video VAE, and MMDiT backbone for streaming video editing.
  • High-throughput 720p deployment. Reaches 30.19 FPS end-to-end throughput at 720x1280 with bounded KV-state inference and stable per-chunk compute.

🚧 TODO

  • Consumer GPU support. Optimize deployment for consumer-grade GPUs such as GeForce RTX 5090.
  • Stronger model version in progress. A more powerful version is under active development, with a particular focus on advancing reference-image-guided video editing (RV2V) capabilities.
  • Release full training and data pipelines. Open-source the complete training framework and data generation pipeline.

🎬 Showcase

JoyAI-Video-Edit is designed for broad video editing tasks, including global appearance changes, local object edits, subject add/remove/replace, background replacement, style transfer, and reference-guided edits.

demo.mp4
Source Prompt Edited
Case 01 source Transform the people, hairstyles, and interior into a British castle aristocratic style. Case 01 edited
Case 02 source Turn the video into a watercolor wash style. Case 02 edited
Case 03 source Make all dogs white, add colorful hats, and turn the sunglasses hot pink. Case 03 edited
Case 04 source Dress the girl in a brown down jacket and blue baseball cap. Case 04 edited
Case 05 source Remove the two white cats in pink clothes on both sides. Case 05 edited

📦 Model Download

Download the released JoyAI-Video-Edit weights from Hugging Face, then place them under:

deploy/deps/checkpoints/JoyAI-Video-Edit/
|-- dit/
|   `-- joyai_video_edit_dit_0804.pth
`-- vae/
    |-- config.json
    `-- diffusion_pytorch_model.safetensors

🚀 Quick Start

1. Install

conda create -n joyai-video-edit python=3.10 -y
conda activate joyai-video-edit
python -m pip install -r requirements.txt

requirements.txt targets NVIDIA CUDA. For AMD ROCm, activate a compatible ROCm PyTorch environment, install requirements-rocm.txt, and build the gfx950 kernels as described in DEPLOYMENT.md. The tested MI350X live preset uses two denoising steps at 1248x720 and a hybrid clean-KV policy: all 40 layers for the permanent global sink once and 24 layers for later bounded tail chunks. Its dynamic full-graph DiT compile can take several minutes at startup, then runs entirely on the selected logical cuda:0; no model role is offloaded to CPU or another GPU. A safe drain/reset keeps real scene cuts from being blended with prior causal state. The one-GPU backend sustained more than 30 FPS while preserving every complete temporal chunk; prompt changes retain that safe teardown while cancelling obsolete queued work and preserving buffered playback. Late browser WebCodecs callbacks are fenced at each prompt boundary, and unused allocator blocks are released after the synchronized teardown so repeated prompts neither corrupt the next H.264 session nor accumulate cached HBM. Model tensors remain on the same GPU. Reference-image shapes are warmed at startup. The browser retains the upstream 24 FPS cap for operating headroom. See HANDOFF.md for the exact TheRock setup, launch, benchmark, and rocprof commands.

2. Prepare Checkpoints

Download the released weights from the Hugging Face link above. MiMo-VL and the ONNX detector files are external runtime dependencies; see DEPLOYMENT.md for deployment details.

3. Launch

cd deploy
bash run_server.sh

Then open:

http://localhost:8080

For remote machines, bind the server to 0.0.0.0. Browser camera capture requires a trusted HTTPS/WSS origin, so use client-resolvable DNS with a trusted TLS reverse proxy or access it as localhost through SSH forwarding; a plain remote HTTP URL can display the UI but cannot obtain camera permission.

🛠️ Custom Deployment

For custom deployment, edit deploy/run_server.sh to set checkpoint paths, accelerator device placement, host, and port. The default script also sets persistent TorchInductor, Triton, and GPU compile-cache directories so compile artifacts are reused across launches.

TORCHINDUCTOR_AUTOGRAD_CACHE is not required for inference-only serving.

📚 Citation

If JoyAI-Video-Edit is useful for your research or product prototype, please cite:

@article{xiao2026joyai,
  title={JoyAI-Video-Edit: Real-Time Open-Ended Video Editing with Autoregressive Diffusion},
  author={Xiao, Yicheng and Dai, Wenxun and Qin, Xinran and Song, Lin and Zhang, Maoquan and Xu, Hang and Chen, Yukang and Li, Yitong and Zhang, Guohui and Zhang, Yuan and Zhang, Xuying and Zhang, Tommy and Yuan, Jianlong and Li, Peihao and Lu, Shuai and Fu, Siming and Zhao, Chuyang and Han, Xin and Huang, Jie and Li, Wenbo and Ma, Guoqing and Huang, Wei and Qi, Xiaojuan and Huang, Haoyang and Duan, Nan},
  journal={arXiv preprint arXiv:2608.03974},
  year={2026}
}

⚖️ License Agreement

JoyAI-Video-Edit is licensed under Apache 2.0.

About

[Official Repo] JoyAI-Video-Edit: Real-Time Open-Ended Video Editing with Autoregressive Diffusion

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages