AIToday
Image GenerationVideo GenerationOpen-Source AIHugging Face BlogPublished: Jul 18, 2026, 01:00 JST

NVIDIA, Hugging Face enable scaled fine-tuning of video and image diffusion models

3 Key Points

  1. What happened

    NVIDIA and Hugging Face announced a collaboration integrating NVIDIA's NeMo Automodel library with Hugging Face's Diffusers, enabling production-grade distributed training of diffusion models (like FLUX.1-dev, Wan 2.1, and HunyuanVideo) directly from the Hugging Face Hub with no checkpoint conversion or model rewrites.

  2. Why it matters

    The integration removes technical barriers for researchers and developers who want to fine-tune or fully train large image and video models. Previously, scaling training required custom scripts and complex conversions; now a single YAML configuration and command-line overrides suffice, and parallelism (FSDP2, tensor, pipeline, context parallel) becomes a configuration choice, not a code rewrite.

  3. What to watch

    The library supports both full fine-tuning and LoRA-style parameter-efficient training, with performance measured on 8× NVIDIA H100 GPUs. For example, FLUX.1-dev full fine-tuning achieves 35.51 ± 1.55 images/s (512×512), while LoRA achieves 53.73 ± 0.48 images/s; HunyuanVideo 1.5 LoRA reaches 1.433 ± 0.006 clips/s on video (512×512×49 frames). The integration is open source under Apache 2.0 and documented in the Diffusers training guide.

Not sure about something? Ask the AI

Questions and answers are published on this page.

Summaries like this, in your inbox every morning.

Context & Analysis

The integration addresses a longstanding friction point in diffusion model training: scaling from research to production requires memory-efficient techniques (FSDP2, tensor parallelism, multiresolution bucketing, latent caching) that were previously scattered across custom training scripts or unavailable entirely. By embedding these capabilities into NeMo Automodel and linking it to Diffusers—the de facto standard interface for open-source diffusion models—NVIDIA and Hugging Face lower the barrier for domain specialization. Researchers can now fine-tune models like FLUX.1-dev (12B) or HunyuanVideo (13B) on single nodes using LoRA or on larger clusters with full fine-tuning, all without model rewrites. The showcase example (fine-tuning FLUX.1-dev on a 78-image Rider–Waite tarot dataset) demonstrates that even small datasets can acquire learned styles—the fine-tuned model's generated images exhibit a cream, red, and black vintage palette with ink contours and allegorical card composition when prompted with the learned trigger token, while the base model remains photographic. Because the recipe stack (FSDP2, checkpointing, generation) carries over unchanged, onboarding a new diffusion model into NeMo Automodel requires only a data preprocessing handler and a model adapter, not a full custom training script. This design choice makes it practical to keep pace with the rapid release cycle of new diffusion models.

FAQ
What models can I fine-tune with NeMo Automodel?
NeMo Automodel currently supports flow-matching models, including FLUX.1-dev, Qwen-Image, Wan 2.1, Wan 2.2, and HunyuanVideo. Ready-to-use fine-tuning recipes are provided in the examples/diffusion/finetune folder.
Do I need to convert checkpoints to use this?
No. Pretrained weights from the Hugging Face Hub work directly. Fine-tuned checkpoints load cleanly back into a DiffusionPipeline for inference or back to the Hub for sharing, with no separate training format conversion required.
What parallelism options are available?
You can switch between FSDP2, tensor parallel, expert parallel, context parallel, and pipeline parallel by declaring configurations, rather than rewriting model code.
Hugging Face BlogRead Original Article

AI news that matters for your work, delivered every morning.

Pick your industry and the AI tools you use, and get news related to your work every day.

Free · 30 seconds with Google · unsubscribe anytimeWhat is AIToday? →

Ask AI

Ask AI anything about this article. The AI reads this article, earlier AIToday articles, and Wikipedia, and cites its sources. Q&As are published on this page for other readers too.

Questions and answers are published on this page.

Related Articles

Next articlePatreon blocks AI training bots with Cloudflare, ends polite requests