LoRA Training Script for Image Editing with FLUX Kontext

Is your feature request related to a problem? Please describe.
Currently, the provided fine-tuning script for FLUX Kontext (train_dreambooth_lora_flux_kontext.py) focuses exclusively on T2I. This is no example to follow to fine-tune the model for instruction-based image editing tasks.

Describe the solution you'd like.
I would like an official example script for LoRA fine-tuning of FLUX Kontext specifically for image editing.

Additional context.
I'd like to express my appreciation for the T2I fine-tuning script—it's a fantastic resource!
Given FLUX Kontext's powerful capabilities, enabling straightforward fine-tuning for image editing would unlock many new applications for the community.

The current T2I script for reference:
examples/dreambooth/train_dreambooth_lora_flux_kontext.py

Inizia leggendo examples/dreambooth/train_dreambooth_lora_flux_kontext.py per comprendere l’esempio esistente di fine-tuning T2I di FLUX Kontext. Ricerca i requisiti per l’editing di immagini basato su istruzioni e confrontali con lo script attuale. Il lavoro è completato quando è disponibile un esempio ufficiale di training di LoRA per l’editing di immagini con FLUX Kontext, con indicazioni d’uso appropriate e validazione.