跳到正文
Hugging Face Blog·· 2023-03-24精选AI 评分62

用 diffusers 训练自己的 ControlNet

Train your ControlNet with diffusers

AI 导读

Hugging Face 发布教程,讲解如何用 diffusers 训练 ControlNet,并以人脸姿态为例走完条件设计、数据集构建和模型训练三步。数据集基于微软 FaceSynthetics 的 100K 合成人脸,用 SPIGA 提取 iBUG 68 点关键点生成条件图,再用 BLIP 生成 caption。

推荐理由

Hugging Face 官方复盘了从条件设计、数据集构建到 diffusers 训练脚本的完整 ControlNet 训练流程,并给出不同显存下的参数配置。

来源:Hugging Face Blog · huggingface.co