使用 kohya-ss 的 train_network.py, train_data_dir 的資料, 要放在 sd-scripts 的 repo 還是 stable-diffusion-webui repo?

train_data_dir 不需要放在特定 repo,你可以放在電腦任何地方,然後在執行 train_network.py 時指定路徑。


📌 1. 兩個 repo (sd-scripts vs stable-diffusion-webui) 的區別

  • sd-scripts 👉 這是 kohya-ss 的 LoRA 訓練腳本,專門用來訓練 LoRA、DreamBooth 等
  • stable-diffusion-webui 👉 這是 AUTOMATIC1111 WebUI,主要用來加載和使用 LoRA,但不負責訓練。

如果你要訓練 LoRA,你應該使用 sd-scripts,並在 train_network.py 指定 train_data_dir
🚫 不要放在 stable-diffusion-webui 裡,因為它不負責 LoRA 訓練


📌 2. train_data_dir 的正確位置

你可以把 datasets/zenmaru_dataset/ 放在電腦的任何地方,例如:

C:\AI\datasets\zenmaru_dataset\

/home/user/AI/datasets/zenmaru_dataset/

訓練指令:

python train_network.py \
  --pretrained_model_name_or_path="C:/AI/Models/stable-diffusion-2-1/" \
  --train_data_dir="C:/AI/datasets/zenmaru_dataset/" \
  --output_dir="C:/AI/output/zenmaru_lora/" \
  --resolution=512,512 \
  --network_dim=32 \
  --learning_rate=1e-4 \
  --train_batch_size=2 \
  --max_train_steps=2000 \
  --lr_scheduler="cosine" \
  --use_8bit_adam

📌 3. 如果還沒下載 sd-scripts

如果你還沒下載 sd-scripts(kohya-ss),請先執行:

git clone https://github.com/kohya-ss/sd-scripts.git
cd sd-scripts
pip install -r requirements.txt

然後 訓練 LoRA 時,不要把資料放進 sd-scripts 內部,你只要在指令裡指定 train_data_dir 就行。


✅ 總結

  • train_data_dir 可以放在電腦任何地方,不需要放進 sd-scriptsstable-diffusion-webui
  • 使用 train_network.py 訓練 LoRA(在 sd-scripts 執行)。
  • stable-diffusion-webui 只用來載入 LoRA,不能訓練

發佈留言

發佈留言必須填寫的電子郵件地址不會公開。 必填欄位標示為 *