🐶

PaperSpaceでkohya版LoRAを動かす方法

2023/02/16に公開

はじめに

https://console.paperspace.com/signup?R=ZS1N2LM
このリンクからPaperspaceを始めたら、10ドルのクレジットが貰えます。

https://onlinegamernikki.com/paperspace_kohyaver_lora_tutorial
この記事を見てやりました。
自分なりにコードを変更して行いました。

備忘録として記録目的で書いたため、分かりにくい部分があるかと思います。

paperspaceの導入、起動までは、参考サイトで画像付きで解説されているため、
そちらで見られると分かりやすいかと思います。

PaperSpace Proを契約

ややこしいのですが、従量課金のものと月額のものがあります。
従量課金は一時間いくらみたいな感じで使えます。
月額のものを契約すると、gpuのところにfreeと表示されます。

Proは月額$8で使えます。GPUはvramが24GBまでのものなら、
6時間刻みで何回でも使えます。

※GPUに空きがなければ使えないこともあります
GPUとプランの早見表です。
GPU
引用元https://qiita.com/kunishou/items/dccb44848e5b572619bc

https://console.paperspace.com/signup?R=ZS1N2LM
このリンクからPaperspaceを始めたら10ドルのクレジットが貰えます。

プロジェクトの作成、起動

paperspaceにログインして、以下のサイトを参考にして進めてください。
gradientで進めていきます。
間違ってcoreで進めないようお気をつけください。
https://onlinegamernikki.com/paperspace_kohyaver_lora_tutorial#toc3

LoRAの導入

WARNING: Running pip as the 'root' user can result in broken permissions and >conflicting behaviour with the system package manager. It is recommended to >use a virtual environment instead: https://pip.pypa.io/warnings/venv

pip時に表示されるこれは気にしなくて大丈夫です。

mkdir LoRA
cd LoRA
sudo apt update -y && sudo apt upgrade -y
git clone https://github.com/kohya-ss/sd-scripts.git
git clone https://github.com/derrian-distro/LoRA_Easy_Training_Scripts.git
cd sd-scripts
apt -y install python3.10
pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116
pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
pip install -U -r requirements.txt
pip install -U --pre triton
pip install -U -I --no-deps xformers-0.0.14.dev0-cp310-cp310-linux_x86_64.whl

//この部分はOSErrorが出てインストールできなかった。LoRAを行う目的の場合、やらなくてOKかも
cd ..
accelerate config
//- This machine (enter)
//- No distributed training (enter)
//- NO (入力)
//- NO
//- NO
//- all
//- fp16(enter)

cp LoRA_Easy_Training_Scripts/lora_train_command_line.py sd-scripts
cd sd-scripts

これで環境構築が終了です。あとはLoRAを実行するだけです。
accelerate launchで実行できます。
僕は以下のように実行しています。
「ファイルパス」の部分は書き換えてください。

accelerate launch --num_cpu_threads_per_process 12 train_network.py --pretrained_model_name_or_path=「ファイルパス.cpkt」 --train_data_dir=「ファイルパス」 --reg_data_dir=「ファイルパス」 --output_dir=「ファイルパス」--resolution=320,960 --train_batch_size=4 --learning_rate=8e-5 --max_train_epochs=10 --save_every_n_epochs=1 --save_model_as=safetensors --clip_skip=2 --seed=42 --color_aug --network_module=networks.lora --keep_tokens=7 --enable_bucket

上記の場合、各自で変更しなければならない部分は4つあります。

  1. –pretrained_model_name_or_path=
    追加学習を行いたいモデルのcpktファイルまでのファイルパスです。

記入例

–pretrained_model_name_or_path=/notebooks/stable-diffusion-webui/models/Stable-diffusion/Evt_M.ckpt
  1. –train_data_dir=
    学習用素材のフォルダを指定します。

記入例

–train_data_dir=/notebooks/LoRA/Training

フォルダ名には命名規則があるので、詳しくはこちらをご覧ください。
https://note.com/kohya_ss/n/nba4eceaa4594#578bd471-5d68-4cc8-bb45-9ca8d80ef1ed

  1. –reg_data_dir=
    正則化画像のフォルダを指定します

記入例

–reg_data_dir=/notebooks/LoRA/reg
  1. –output_dir=
    学習結果を保存するフォルダを指定します。

記入例

–output_dir=/notebooks/LoRA/testLoRA

2回目以降の起動

cd LoRA
sudo apt update -y && sudo apt upgrade -y
cd sd-scripts
apt -y install python3.10
pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116
pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
pip install -U -r requirements.txt
pip install -U --pre triton
pip install -U -I --no-deps xformers-0.0.14.dev0-cp310-cp310-linux_x86_64.whl
//この部分はOSErrorが出てインストールできなかった。LoRAを行う目的の場合、やらなくてOKかも
cd ..
accelerate config
cd sd-scripts

画像生成のおすすめ本

【2023最新】1000種類以上のプロンプト収録! NovelAIでもローカルでも使える呪文集: 小ネタ・テクニック・おすすめツールも紹介 (AI絵師の入門書) Kindle版

https://amzn.to/3YVp90y

AIとコラボして神絵師になる 論文から読み解くStable Diffusion
https://amzn.to/41aLQ30

さきがけ技術 AUTOMATIC1111 / Stable Diffusion web UI の使い方
https://amzn.to/3Kcjboe

Discussion