NVIDIA与Hugging Face联手，规模化微调视频与图像模型
技术· 原作者：Hugging Face· 2 分钟阅读0 阅读
NVIDIA NeMo Automodel 与 Hugging Face Diffusers 集成，提供自动化、可扩展的微调方案，支持视频和图像模型在大规模集群上高效训练，降低开发门槛。
NVIDIA 与 Hugging Face 宣布，NVIDIA NeMo Automodel 现已集成 🤗 Diffusers 库，为视频和图像生成模型的微调提供端到端自动化方案。这一集成使得开发者能够在 NVIDIA GPU 集群上，以极简配置完成从数据预处理到分布式训练的全流程。
NeMo Automodel 自动处理模型并行、数据加载和超参数调优，而 Diffusers 则提供丰富的预训练模型库（如 Stable Diffusion、VideoCrafter 等）。用户只需几行代码即可启动训练，无需手动管理基础设施。
该方案特别适合需要大规模微调的场景，例如企业定制化图像生成、视频内容风格迁移以及长视频生成。NVIDIA 表示，结合 NeMo 的自动混合精度和梯度累积策略，训练速度可提升 30% 以上，同时显存占用降低 20%。
目前，NVIDIA NeMo Automodel 与 Diffusers 的集成已在 NVIDIA AI Enterprise 中可用。Hugging Face 社区用户也可通过 accelerate 和 torchrun 快速接入，详情可参考 官方博客。
原文链接：Hugging Face
本文由前途科技编辑整理
想了解 AI 如何助力您的企业？
免费获取企业 AI 成熟度诊断报告，发现转型机会
//
24小时热榜
热门标签
关注公众号
扫码关注，获取最新 AI 资讯
免费获取 AI 落地指南
3 步完成企业诊断，获取专属转型建议
已有 200+ 企业完成诊断