How to train a FLUX lora?

[Xuan-World]

I don't want to train a dreambooth lora or something like that. I just want to train a lora on FLUX-schnell. So which script should I refer to in ''examples'' directory?

[bmaxdk]

train_text_to_image_lora.py in the examples directory, will be the one you looking for since it is designed specifically for training LoRA models without involving DreamBooth.

python train_text_to_image_lora.py \
  --pretrained_model_name_or_path="path_or_identifier_to_FLUX-schnell" \ # Path or Hugging Face identifier for FLUX-schnell
  --train_data_dir=".......
  # add other cofig

[Xuan-World]

thanks a lot.

[Doby-Xu]

Hey guys. Have you ever tried to run that script? I don't think that would work, since the implementation in the example is completely based on Stabel Diffusion, instead of FLUX.

[Xuan-World]

Yes, it's really tough to change that script to adapt to FLUX. I'm trying on it for several days. A Suggestion: refer to the implementation script of DreamBooth FLUX LoRA and Control-FLUX LoRA.

[asomoza]

you will have an easier time with the other way around: start with the dreamBooth FLUX LoRA script and remove all the parts that are for dreambooth.