From 8d48ad4e85874b3dca610a659e3e70233bc38ebc Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Sun, 11 Aug 2024 10:28:39 -0600 Subject: [PATCH] fixed bug I added to demo config --- README.md | 2 +- config/examples/train_lora_flux_24gb.yaml | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/README.md b/README.md index d2c21dc4..f60597cc 100644 --- a/README.md +++ b/README.md @@ -39,7 +39,7 @@ cd ai-toolkit git submodule update --init --recursive python -m venv venv .\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 ``` diff --git a/config/examples/train_lora_flux_24gb.yaml b/config/examples/train_lora_flux_24gb.yaml index 74eceae9..8f862848 100644 --- a/config/examples/train_lora_flux_24gb.yaml +++ b/config/examples/train_lora_flux_24gb.yaml @@ -31,7 +31,6 @@ config: shuffle_tokens: false # shuffle caption order, split by commas cache_latents_to_disk: true # leave this true unless you know what you're doing resolution: [ 512, 768, 1024 ] # flux enjoys multiple resolutions - num_workers: 0 train: batch_size: 1 steps: 4000 # total number of steps to train 500 - 4000 is a good range