ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-01-26 16:39:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixed bug I added to demo config

Browse Source
This commit is contained in:
Jaret Burkett
2024-08-11 10:28:39 -06:00
parent ec1ea7aa0e
commit 8d48ad4e85
2 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -39,7 +39,7 @@ cd ai-toolkit
git submodule update --init --recursive git submodule update --init --recursive
python -m venv venv python -m venv venv
.\venv\Scripts\activate .\venv\Scripts\activate
pip install torch --use-pep517 --extra-index-url https://download.pytorch.org/whl/cu118 pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121
pip install -r requirements.txt pip install -r requirements.txt
``` ```

1
config/examples/train_lora_flux_24gb.yaml
View File

@@ -31,7 +31,6 @@ config:
shuffle_tokens: false # shuffle caption order, split by commas shuffle_tokens: false # shuffle caption order, split by commas
cache_latents_to_disk: true # leave this true unless you know what you're doing cache_latents_to_disk: true # leave this true unless you know what you're doing
resolution: [ 512, 768, 1024 ] # flux enjoys multiple resolutions resolution: [ 512, 768, 1024 ] # flux enjoys multiple resolutions
num_workers: 0
train: train:
batch_size: 1 batch_size: 1
steps: 4000 # total number of steps to train 500 - 4000 is a good range steps: 4000 # total number of steps to train 500 - 4000 is a good range