gen_img_diffusers

Last-modified: 2023-11-09 (木) 21:06:59

sd-scriptsでできる画像生成スクリプトのこと。

sd-scriptsにおける画像生成ツール

概要

使い方

  • ほぼReadmeに書いてあるので、そちらを参照。
  • 以下、実行用batの作成例
@echo off

setlocal enabledelayedexpansion

rem sd-scriptsの場所
set sd_path="C:\(個々の環境に合わせて入力)\sd-scripts"

rem 事前学習モデル
set ckpt_file="C:\(ベースとなるcheckpointのパスを指定)hogehoge.safetensors"

rem 生成画像の出力先
set output_path="C:\(任意のフォルダを指定)\Output_gen_img_diffusers"

rem CPUスレッド数
set cpu_thread=6


rem ------------------------------------------------------------

rem 学習実行
cd /d %sd_path%
call venv\Scripts\activate.bat & call :main & pause & exit

:main

accelerate launch --num_cpu_threads_per_process %cpu_thread% gen_img_diffusers.py ^
	--ckpt %ckpt_file% ^
	--outdir %output_path% ^
    --xformers --fp16 ^
	--W 512 --H 704 ^
	--scale 12.5 ^
	--sampler k_euler_a ^
    --steps 32 ^
	--batch_size 1 ^
	--images_per_prompt 2 ^
    --from_file "C:\(プロンプトを記載したテキストファイルの絶対パスを記入のこと).txt" ^
	--network_module networks.lora ^
	--network_weights "C:\(LoRAモデルの絶対パスをここへ入力。使わない場合は不要なので行を削除すること)"

exit /b
  • LyCORISモジュール(lycoris.kohya)は「完全には」使えない。(一部アルゴリズムは使えるかも?)
    • なぜなら、LoRA学習時とは違って画像生成を行う際には、専用の関数定義を行う行程がsd-scriptsコード内部に存在しているため。