| accelerate launch train_dreambooth_lora_flux_kontext.py \ | |
| --pretrained_model_name_or_path=black-forest-labs/FLUX.1-Kontext-dev \ | |
| --dataset_name="polytechXhf/onepiece-dataset" \ | |
| --output_dir="kontext-OnePiece" \ | |
| --image_column="image" \ | |
| --caption_column="text" \ | |
| --validation_prompt="a person in fighting mode in OnePiece style" \ | |
| --rank=32 \ | |
| --lora_alpha=32 \ | |
| --train_text_encoder \ | |
| --lora_layers="attn.to_k,attn.to_q,attn.to_v,attn.to_out.0,attn.to_add_out,ff.net.0.proj,ff.net.2,proj_mlp" \ | |
| --mixed_precision="bf16" \ | |
| --instance_prompt="in OnePiece style" \ | |
| --resolution=1024 \ | |
| --train_batch_size=1 \ | |
| --guidance_scale=1 \ | |
| --gradient_accumulation_steps=1 \ | |
| --gradient_checkpointing \ | |
| --optimizer="adamw" \ | |
| --use_8bit_adam \ | |
| --cache_latents \ | |
| --learning_rate=1e-4 \ | |
| --lr_scheduler="constant" \ | |
| --lr_warmup_steps=0 \ | |
| --max_train_steps=1000 \ | |
| --seed="0" \ | |
| --push_to_hub | |
| accelerate launch train_dreambooth_lora_flux_kontext.py --pretrained_model_name_or_path=black-forest-labs/FLUX.1-Kontext-dev --dataset_name="polytechXhf/onepiece-dataset" --output_dir="kontext-OnePiece" --image_column="image" --caption_column="text" --validation_prompt="a person in fighting mode in OnePiece style" --rank=32 --lora_alpha=32 --train_text_encoder --lora_layers="attn.to_k,attn.to_q,attn.to_v,attn.to_out.0,attn.to_add_out,ff.net.0.proj,ff.net.2,proj_mlp" --mixed_precision="bf16" --instance_prompt="in OnePiece style" | |
| --resolution=1024 --train_batch_size=1 --guidance_scale=1 --gradient_accumulation_steps=1 --gradient_checkpointing --optimizer="adamw" --use_8bit_adam --learning_rate=1e-4 --lr_scheduler=" | |
| constant" --max_train_steps=2000 --seed="0" --push_to_hub |