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-scripts
或stable-diffusion-webui
。- 使用
train_network.py
訓練 LoRA(在sd-scripts
執行)。 stable-diffusion-webui
只用來載入 LoRA,不能訓練。