文章预览
推荐阅读列表: 扩散模型实战(一):基本原理介绍 扩散模型实战(二):扩散模型的发展 扩散模型实战(三):扩散模型的应用 扩散模型实战(四):从零构建扩散模型 扩散模型实战(五):采样过程 扩散模型实战(六):Diffusers DDPM初探 在本文中,我们以生成绚丽多彩的蝴蝶图像为例,学习Diffusers库相关知识,并学会训练自己的扩散模型。 一、环境准备工作 1.1 安装Diffusers库 % pip install -qq -U diffusers datasets transformers accelerate ftfy pyarrow==9.0.0 1.2 登录huggingface(optional) 我们计划把训练好的模型上传到huggingface中,因此我们需要首先登录huggingface,可以通过访问https://huggingface.co/settings/tokens获取huggingface的token。 复制上图的token,然后执行下面的代码,并粘贴到执行如下代码的文本框中: from huggingface_hub import notebook_login notebook_lo
………………………………