M_sunglasses / config.yaml
Sri2901's picture
Upload model via CLI
09d171b verified
job: extension
config:
name: M_sunglasses
process:
- type: ui_trainer
training_folder: /workspace/ai-toolkit/output
sqlite_db_path: /workspace/ai-toolkit/aitk_db.db
device: cuda
trigger_word: s@ngl@ss
performance_log_every: 10
network:
type: lora
linear: 32
linear_alpha: 32
conv: 16
conv_alpha: 16
lokr_full_rank: true
lokr_factor: -1
network_kwargs:
ignore_if_contains: []
save:
dtype: bf16
save_every: 250
max_step_saves_to_keep: 8
save_format: diffusers
push_to_hub: false
datasets:
- folder_path: /workspace/ai-toolkit/datasets/msunglasses
control_path: null
mask_path: null
mask_min_value: 0.1
default_caption: ''
caption_ext: txt
caption_dropout_rate: 0.05
cache_latents_to_disk: false
is_reg: false
network_weight: 1
resolution:
- 512
- 768
- 1024
- 1536
- 1280
controls: []
shrink_video_to_frames: true
num_frames: 1
do_i2v: true
flip_x: false
flip_y: false
train:
batch_size: 1
bypass_guidance_embedding: false
steps: 2000
gradient_accumulation: 1
train_unet: true
train_text_encoder: false
gradient_checkpointing: true
noise_scheduler: flowmatch
optimizer: adamw8bit
timestep_type: sigmoid
content_or_style: balanced
optimizer_params:
weight_decay: 0.0001
unload_text_encoder: false
cache_text_embeddings: false
lr: 0.0002
ema_config:
use_ema: true
ema_decay: 0.99
skip_first_sample: false
disable_sampling: false
dtype: bf16
diff_output_preservation: false
diff_output_preservation_multiplier: 1
diff_output_preservation_class: person
switch_boundary_every: 1
model:
name_or_path: black-forest-labs/FLUX.1-dev
quantize: false
qtype: qfloat8
quantize_te: true
qtype_te: qfloat8
arch: flux
low_vram: false
model_kwargs: {}
sample:
sampler: flowmatch
sample_every: 250
width: 1024
height: 1024
samples:
- prompt: s@ngl@ss, A young man in a close-up portrait, styled in a muted olive
crewneck sweater. He wears clear-frame glasses that highlight his natural
features, while his slightly tilted head and contemplative gaze add depth
and character to the shot. Clean, minimal composition keeps the focus on
the eyewear and expression. studio BG
- prompt: s@ngl@ss, A young man in a centered, front-facing pose, wearing bold
square-frame black sunglasses that command attention. His ribbed black crewneck
sweater enhances the sleek, monochrome aesthetic, while the smooth gradient
backdrop highlights the structured silhouette. The composition radiates
confidence and modern minimalism. studio bg
- prompt: s@ngl@ss, A young man in a three-quarter side pose, wearing square
tortoiseshell sunglasses with gradient brown lenses. The neutral background
softly contrasts with his earthy-toned trench coat layered over a cream
knit, creating a refined yet relaxed composition. His upward gaze adds a
sense of effortless sophistication.
neg: ''
seed: 42
walk_seed: true
guidance_scale: 3
sample_steps: 30
num_frames: 1
fps: 1
meta:
name: M_sunglasses
version: '1.0'