Hướng dẫn train lora flux1 với fluxgym

Yêu cầu:

• Card đồ họa từ 12G VRAM trở lên
• Ổ cứng có dung lượng trống từ 40GB trở lên
• Trên máy đã cài đặt Python 3.10, Git, Pip. Nếu chưa cài xin xem lại video:

Cách cài đặt stable diffusion tiết kiệm ổ cứng nhất bạn cần biết

Các bước cài đặt (trên Windows)

Từ cửa sổ cmd, chuyển đến thư mục muốn cài fluxgym, gõ lần lượt các lệnh sau:

git clone https://github.com/cocktailpeanut/fluxgym

cd fluxgym

git clone -b sd3 https://github.com/kohya-ss/sd-scripts

python -m venv env

env\Scripts\activate

cd sd-scripts

pip install -r requirements.txt

cd ..

pip install -r requirements.txt

pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

Chạy fluxgym:

python app.py

(Từ các lần tiếp theo trở đi, copy file RUN.bat vào thư mục fluxgym từ thư mục chia sẻ trên google drive để chạy cho nhanh)

Chuẩn bị dữ liệu

• Chuẩn bị từ 4-20 ảnh, kích thước từ 512px trở lên, tốt nhất từ 1024px
• Nên dùng ảnh có tỷ lệ 1:1
• Ảnh sắc nét, không có watermark

Tải ảnh lên Fluxgym và train

• Cấu hình train
  • Repeat trains per image: Số lần lặp trên mỗi hình ảnh
  • Max Train Epochs: Số lần train
  • Expected training steps (Số bước train): = Repeat trains per image x Max Train Epochs
  • Sample Image Every N Steps: Tạo ảnh mẫu sau mỗi N bước train

Prompt mẫu:

carved_pumpkin, Celebrity Taylor Swift carved into a pumpkin, An abstract representation of an autumn-themed wine tasting scene. Bold, fluid shapes and soft curves depict a rustic outdoor garden or terrace, with muted golden and orange hues of fall. Hints of wine glasses and bottles blend into swirling autumn leaves and abstracted pumpkins, while string lights appear as soft, glowing orbs. 

carved_pumpkin, Celebrity Gordon Ramsey carved into a pumpkin

carved_pumpkin, Celebrity Elon musk carved in to a pumpkin, pumpkin in the hands of a blonde girl, standing in the middle of a pumpkin field, DOF

carved_pumpkin, A man is carving Mr. Obama on a pumpkin, in a basement room, dark scene, with hall decorations, DOF

Chi tiết hướng dẫn tại video:

HMT*