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.
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.
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.
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 →
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.
For example, today's edition would include:
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 anything about this article. Q&As are published on this page for other readers too.
Hugging Face released @huggingface/kernels, a library for running optimized WebGPU kernels from the Hugging Fa…

Visko raised $10 million in pre-seed funding from Llama Ventures and opened public access to its first foundat…
Israeli startup DataAgent Ltd
Chinese large-model developer Z.ai says it can now support large-scale inference using roughly 100,000 domesti…

Recent controversies include Ajinomoto's official X account posting an AI-edited image and a restaurant menu s…

Broadcom announced VMware AI Factory, a software-defined foundation for VMware Private AI Cloud, at VMware Exp…
