human-hig-flux / config.yaml
linhqyy's picture
Upload folder using huggingface_hub
3a0b9a1 verified
job: extension
config:
name: human
process:
- type: sd_trainer
training_folder: output
device: cuda:0
network:
type: lora
linear: 16
linear_alpha: 16
save:
dtype: float16
save_every: 1000
max_step_saves_to_keep: 4
datasets:
- folder_path: /workspace/imgs
caption_ext: txt
caption_dropout_rate: 0.05
shuffle_tokens: false
cache_latents_to_disk: true
resolution:
- 1024
train:
batch_size: 1
steps: 10000
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: false
sample:
sampler: flowmatch
sample_every: 500
width: 1344
height: 768
prompts:
- An illustration of, a young woman, lying down, with long, flowing brown hair,
wearing a white dress, eyes closed, lips slightly parted, serene and peaceful
expression, hand resting behind head, soft ambient lighting, night sky, background
with blurred, dark structures, hints of nature and foliage, bright moonlight
casting a gentle glow, realistic yet artistic style, high-resolution, extremely
detailed.
neg: ''
seed: 42
walk_seed: true
guidance_scale: 4
sample_steps: 20
meta:
name: human
version: '1.0'