lora_task337 / config.yaml
vBogdan's picture
Upload folder using huggingface_hub
3e4c23d verified
config:
name: job_task337
process:
- datasets:
- cache_latents_to_disk: true
caption_dropout_rate: 0.05
caption_ext: txt
folder_path: /app/datasets/job_task337
resolution:
- 512
- 768
- 1024
device: cuda:0
model:
is_flux: true
name_or_path: black-forest-labs/FLUX.1-dev
quantize: false
network:
linear: 16
linear_alpha: 16
type: lora
sample:
guidance_scale: 4
height: 1024
prompts:
- digital anime-style drawing of a red-haired girl with fair skin standing on
a sunny beach, wearing a white off-shoulder crop top and high-waisted shorts,
blue ocean and clear sky in the background, white clouds, green grass in the
distance, she is barefoot, confident expression, slender and curvy physique,
long hair with bangs, bright sunlight, detailed shading, realistic textures,
realistic proportions, standing, legs together, feet touching sand
sample_every: 200
sample_steps: 20
sampler: flowmatch
seed: 42
width: 1024
save:
dtype: float16
max_step_saves_to_keep: 2
save_every: 200
train:
batch_size: 1
dtype: bf16
ema_config:
ema_decay: 0.99
use_ema: true
gradient_accumulation_steps: 4
gradient_checkpointing: false
lr: 0.0002
noise_scheduler: flowmatch
optimizer: adamw
steps: 800
train_text_encoder: false
train_unet: true
training_folder: output
type: sd_trainer
job: extension