Update README.md
Browse files
README.md
CHANGED
|
@@ -1,2 +1,46 @@
|
|
| 1 |
-
|
| 2 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# frog train benchmark
|
| 2 |
+
|
| 3 |
+
ハローカエル(なぜかカエルが出ない)
|
| 4 |
+
|
| 5 |
+
## 概要
|
| 6 |
+
|
| 7 |
+
- kohya train_networkのベンチです
|
| 8 |
+
- バッチをダウンロードして任意のディレクトリ(日本語やスペースが無いと良い)に置いて実行します
|
| 9 |
+
- https://huggingface.co/aka7774/frog_bench/resolve/main/frog_bench.bat
|
| 10 |
+
- Windows以外ではサンプル通りに学習を実行してtimeを計測してください
|
| 11 |
+
- サンプルのカエルを使います
|
| 12 |
+
- https://note.com/kohya_ss/n/nb20c5187e15a
|
| 13 |
+
- https://note.com/api/v2/attachments/download/e3cd9aa39e600cac51e2022eaa01a931
|
| 14 |
+
- 中身をこのリポジトリにコピーしてあります
|
| 15 |
+
- モデルはSDv1.5を使います
|
| 16 |
+
- https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned.safetensors
|
| 17 |
+
- pruneしたファイルを用意してあります
|
| 18 |
+
|
| 19 |
+
## 設定
|
| 20 |
+
|
| 21 |
+
バッチファイルを編集することでいくつかの設定が出来ます。
|
| 22 |
+
|
| 23 |
+
- bypass Install CUDA Toolkit
|
| 24 |
+
- pytorchのlibにPATHを通すことでCUDA Toolkitのインストールを省略する
|
| 25 |
+
- Path to
|
| 26 |
+
- PythonとgitにPATHが通っていない時にフルパスで指定する
|
| 27 |
+
- Pythonとgit自体のインストールは別途必要
|
| 28 |
+
- VERS
|
| 29 |
+
- 1はkohya推奨バージョン(古い)
|
| 30 |
+
- 2は1111推奨バージョン(新しい) xformersが動かない可能性がある
|
| 31 |
+
- MODE
|
| 32 |
+
- 複数回実行したい時にインストールや学習を飛ばす
|
| 33 |
+
- BATCH_SIZE
|
| 34 |
+
- VRAMが10GB未満の時に減らす
|
| 35 |
+
|
| 36 |
+
# 変更点
|
| 37 |
+
|
| 38 |
+
- num_cpu_threads_per_process(未変更)
|
| 39 |
+
- 1のほうがいいらしいけどサンプルが4なのでそのまま
|
| 40 |
+
- learning_rate
|
| 41 |
+
- 途中で仕様変更があったので1桁減らしたほうがいいらしいけどそのまま
|
| 42 |
+
- inference(gen_img_diffusers.py)
|
| 43 |
+
- 公式のオプションがbf16だったけどA100でしか動かないというエラーが出るのでfp16に変えてあります
|
| 44 |
+
- interactiveしなくて済むようにした
|
| 45 |
+
- accelerate config
|
| 46 |
+
- accelerate launchに引数を追加することで割愛しています
|