svjack / stable-diffusion-pokemon Goto Github PK

在宝可梦数据集（Pokemon-Blip-Captions）的英语、日语、中文版本上微调Stable Diffusion的示例

In English

Stable Diffusion是现在一流的文本转图片生成模型。
现如今，借助于提供多模态能力的diffusers工程，人们可以自定义它们的条件或非条件图像（是否以文本提示作为条件）生成模型。
这个工程的目标是实现diffuser提供的基于lambdalabs/pokemon-blip-captions的text to image example微调任务，并将该任务迁移到日文及中文数据集上进行实践。
通过比较结论将会对Stable Diffusion在不同语言上的微调任务给出指导。 所有的代码都依据官方的train_text_to_image.py进行修改，并使得其对于中文和日文是有效的。 得到的三种语言的训练模型如下：English , Japanese 及 Chinese.

运行install.sh将会安装所有的依赖并下载所有需要的模型（保证您注册并具有huggingface账号以及您的token 下载后可尝试运行run_en_model.py, run_ja_model.py 及 run_zh_model.py

为了在日文及中文领域进行调试，我们需要将lambdalabs/pokemon-blip-captions。我已经使用 DeepL 对其进行翻译并上传至 huggingface dataset hub。分别位于 svjack/pokemon-blip-captions-en-ja 和 svjack/pokemon-blip-captions-en-zh.

英文版本是一个仅仅将train_text_to_image.py的脚本由函数更改为notebook模式，位于train_en_model.py

日文版本使用rinnakk/japanese-stable-diffusion替换预训练模型，位于train_ja_model.py

中文版本使用IDEA-CCNL/Taiyi-CLIP-Roberta-102M-Chinese 替换预训练分词器以及文本编码器，并将BertForTokenClassification的logit输出进行padding来代替CLIPTextModel，位于train_zh_model.py。

为了在推断阶段使得所有结果可见，我关闭了safety_checker。

Prompt	English	Japanese	Chinese
A cartoon character with a potted plant on his head 鉢植えの植物を頭に載せた漫画のキャラクター 一个头上戴着盆栽的卡通人物
cartoon bird 漫画の鳥 卡通鸟
blue dragon illustration ブルードラゴンのイラスト 蓝色的龙图

在英文、日文、中文下的预训练模型分别训练了26000, 26000 及 20000步。
对于训练结果的解释是这样的：rinnakk/japanese-stable-diffusion由于是日文的原生模型，所以含有很多宝可梦的特征。Stable Diffusion在英文领域微调的很好。IDEA-CCNL/Taiyi-CLIP-Roberta-102M-Chinese 如模型卡片所说在中文上能够起到基本的文本表征作用。

svjack - [email protected] - [email protected]

Project Link:https://github.com/svjack/Stable-Diffusion-Pokemon

• Stable Diffusion
• diffusers
• DeepL
• rinnakk/japanese-stable-diffusion
• IDEA-CCNL/Taiyi-CLIP-Roberta-102M-Chinese
• svjack