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

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

Key takeaway

  • NVIDIA and Hugging Face have integrated the NeMo Automodel training library with the Diffusers ecosystem, making it practical for researchers and developers to fine-tune large diffusion models (FLUX.1-dev, Wan 2.1, HunyuanVideo) at scale without checkpoint conversion or custom code rewrites.

  • The collaboration combines NVIDIA's distributed training capabilities—sharding, latent caching, multiresolution bucketing, and parallelism options—with Hugging Face's model hub, so users can point to any Diffusers model on the hub and begin training with existing YAML configs and command-line overrides.

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.

Ask the AI about this article →

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

Get the latest Image Generation news every morning

For example, today's edition would include:

  • Why AI images feel 'cringey' to consumersITmedia AI+ · 14h ago
  • Disney concept art auction fetches $3.43MTop Companies AI · 2d ago
  • ESP32-P4 Reads Water Meter with AIr/robotics · 2d ago

AI-summarized, only the topics you pick — one digest a day via Email, Slack, or Discord.

Free · takes 30 seconds · unsubscribe anytimeWhat is AIToday? →

Ask AI

Ask AI anything about this article. Q&As are published on this page for other readers too.

Related Articles

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