kohya_ss GUI安装教程

kohya_ss GUI安装教程

先附上两张我生成的动漫美女

kohya_ss GUI安装教程_第1张图片

kohya_ss GUI安装教程_第2张图片

介绍:

This repository contains training, generation and utility scripts for Stable Diffusion.

kohyass是用来训练Stable Diffusion模型的。

1.conda创建python虚拟环境

conda create -n kohyass python=3.10 -y
conda activate kohyass

3.clone项目并下载依赖

git clone https://ghproxy.com/https://github.com/kohya-ss/sd-scripts.git
cd sd-scripts
设置清华源
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
安装依赖
pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
pip install --upgrade -r requirements.txt

4.安装xformers(可选)

git clone https://ghproxy.com/https://github.com/facebookresearch/xformers/
cd xformers/

回退到0.0.4dev0版本,保持和kohya-ss使用的windows预编译版本一致,避免可能的兼容问题

git reset --hard 3633e1afc7bffbe61957f04e7bb1a742ee910ace
cd .git/
vi config
替换所有的github链接为加速链接 https://ghproxy.com/
git submodule update --init --recursive
# 强制使用CUDA,不开启可能会导致xformers未和CUDA一起编译
export FORCE_CUDA="1"
# 进入https://developer.nvidia.com/cuda-gpus#compute
# 设置所用显卡对应的Compute Capability,3090和A5000都是8.6
export TORCH_CUDA_ARCH_LIST=7.0
# PATH环境变量中追加cuda目录,确保编译时能识别镜像预置的cuda11.6
export PATH=$PATH:/usr/local/cuda
# 确保gcc编译时能够识别cuda的头文件
export CPATH=/usr/local/cuda/targets/x86_64-linux/include

pip install -r requirements.txt
pip install -e .

在sd-scripts同级目录安装kohya_ss

git clone https://ghproxy.com/https://github.com/bmaltais/kohya_ss.git
pip install --upgrade -r requirements.txt

启动webui

python ./kohya_gui.py --listen 127.0.0.1 --server_port 7860 --inbrowser --share

效果如下:

kohya_ss GUI安装教程_第3张图片

最后欢迎大家猫说AI(wx搜索哦),会持续分享更多免费教程! 你的关注就是我的动力!

公众号还接入了chatgpt哦,免费给大家使用!

你可能感兴趣的:(python,开发语言)