本项目是对 https://github.com/justinpinkney/stable-diffusion 项目的改造 系列介绍文章见
【stable-diffusion企业级教程01】从零开始,目标:成为神奇宝贝训练大师!
【stable-diffusion企业级教程02】pytorch-lighting?我要deepspeed
【stable-diffusion企业级教程03】FP16真香!(想说爱你不容易)
【stable-diffusion企业级教程04】EMA你走,拥抱16G显存!Xformers是未来!
原始pytorch-lighting
python main.py -t --base configs/stable-diffusion/pokemon.yaml --gpus 6,7 --scale_lr False --num_nodes 1 --check_val_every_n_epoch 10 --finetune_from weights/sd-v1-4-full-ema.ckpt
torch
python3 main_torch.py
deepspeed
sh sd_run.sh
-
- pokemon原始数据 链接: https://pan.baidu.com/s/1GDfax9DN4FP1UcHVqfgOlw?pwd=8w2g 提取码: 8w2g 下载后将 .zip 文件放在 <项目目录>/data/ 下并解压,得到 <项目目录>/data/pokemon_data/ 目录
-
- 翻译数据 链接: https://pan.baidu.com/s/1fK8wywhfB-IMYiacU5iBrQ?pwd=wwpp 提取码: wwpp