job: extension | |
config: | |
name: riz | |
process: | |
- type: sd_trainer | |
training_folder: output | |
device: cuda:0 | |
trigger_word: riz ahmed | |
network: | |
type: lora | |
linear: 32 | |
linear_alpha: 32 | |
save: | |
dtype: float16 | |
save_every: 200 | |
max_step_saves_to_keep: 1000 | |
datasets: | |
- folder_path: /opt/LoRA/images/riz | |
caption_ext: txt | |
caption_dropout_rate: 0.05 | |
shuffle_tokens: false | |
cache_latents_to_disk: true | |
resolution: | |
- 512 | |
- 768 | |
- 1024 | |
- 1280 | |
train: | |
batch_size: 1 | |
steps: 4000 | |
gradient_accumulation_steps: 1 | |
train_unet: true | |
train_text_encoder: false | |
gradient_checkpointing: true | |
noise_scheduler: flowmatch | |
optimizer: adamw8bit | |
lr: 0.0001 | |
ema_config: | |
use_ema: true | |
ema_decay: 0.99 | |
dtype: bf16 | |
model: | |
name_or_path: black-forest-labs/FLUX.1-dev | |
is_flux: true | |
quantize: true | |
sample: | |
sampler: flowmatch | |
sample_every: 200 | |
width: 1024 | |
height: 1024 | |
prompts: | |
- A photo of riz ahmed wearing a baggy jacket, no shirt and low rise jeans. | |
The background is a various flora and fauna. The photo was shot during golden | |
hour and has heavy film grain. The photo was shot on a kodak film camera and | |
looks analog style. | |
neg: '' | |
seed: 42 | |
walk_seed: false | |
guidance_scale: 3.5 | |
sample_steps: 20 | |
meta: | |
name: riz | |
version: '1.0' | |