Train Your Own Flux.1 LoRA (Patreon)

Video - https://youtu.be/7AhQcdqnwfs

Flux is a great model with loads of potential, but when it comes to generating images with almost any art style it can struggle a bit. You may also wish to add your own face, a pet or indeed anything you have some images of!

Woodblock art style without a lora...

Woodblock art style WITH a lora :)

Ai-Toolkit Quick Install with Conda

git clone https://github.com/ostris/ai-toolkit.git

cd ai-toolkit

git submodule update --init --recursive

conda create -y --name my-ai-toolkit python=3.11

conda activate my-ai-toolkit

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

pip install -r requirements.txt

For alternative pytorch installs, see https://pytorch.org/

Changes for Microsoft Windows

Users of Microsoft Windows could use this command instead:

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

Also, to install git if you don't have it :

conda install git

Datasets

4 to 40 images should be fine to start with

Use the optional Florence Captioner to create matching text files

Links

Online - https://fal.ai/models/fal-ai/flux-lora-general-training

Local - https://github.com/ostris/ai-toolkit

Anaconda - https://www.anaconda.com/

Miniconda - https://docs.anaconda.com/miniconda/