thu / trainFlux /trainFlux.sh
housearch's picture
Update trainFlux/trainFlux.sh
4317407 verified
#使用 bash指令執行這個腳本 ---> bash trainFlux.sh
#!/bin/bash
cd /workspace
wget --trust-server-name --content-disposition https://huggingface.co/housearch/thu/resolve/main/trainFlux/download_caption_flux.py
wget --trust-server-name --content-disposition https://huggingface.co/housearch/thu/resolve/main/trainFlux/landscapeOverhead_A6000_flux.json
python -m venv env
source /workspace/env/bin/activate
pip install datasets huggingface_hub Pillow
python download_caption_flux.py
deactivate
cd /workspace
git clone https://github.com/bmaltais/kohya_ss
cd kohya_ss
#預設是訓練SD1.5, SDXL的版本,如果要訓練SD3, Flux 必須切換至sd3-flux.1
git branch -a
git checkout sd3-flux.1
./setup-runpod.sh
./gui.sh --share --headless
'''
另開 terminal 視窗,把以下內容貼上並按下enter執行,直接下載訓練Flux LoRA必須的相關模型(節省時間)
wget --trust-server-name --content-disposition https://huggingface.co/realung/flux1-dev.safetensors/resolve/main/flux1-dev.safetensors
wget --trust-server-name --content-disposition https://huggingface.co/comfyanonymous/flux_text_encoders/resolve/main/t5xxl_fp16.safetensors
wget --trust-server-name --content-disposition https://huggingface.co/comfyanonymous/flux_text_encoders/resolve/main/clip_l.safetensors
wget --trust-server-name --content-disposition https://huggingface.co/StableDiffusionVN/Flux/resolve/main/Vae/flux_vae.safetensors
'''
#./gui.sh --share --headless --config "/workspace/config.toml"