wallet_pose / config.yaml
Sri2901's picture
Upload model via CLI
46b343c verified
job: extension
config:
name: wallet-poses
process:
- type: ui_trainer
training_folder: /workspace/ai-toolkit/output
sqlite_db_path: /workspace/ai-toolkit/aitk_db.db
device: cuda
trigger_word: w@llet
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/wallet_poses
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
- 1280
- 1536
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: "w@llet, Man in a beige blazer and white t-shirt, holding a black\
\ leather wallet close to his chin with his right hand, slightly angled\
\ to showcase the wallet\u2019s texture and stitching. The composition is\
\ a tight portrait crop, centering on his face, upper torso, and wallet\
\ as the focal point. His gaze is directed downward with a subtle, thoughtful\
\ expression. The background is plain white with soft gradient lighting,\
\ evenly diffused to eliminate shadows, giving the scene a polished and\
\ minimal aesthetic that emphasizes the wallet\u2019s premium finish."
- prompt: w@llet, Man in a navy blazer and white t-shirt, holding a brown leather
wallet with his right hand, placing it into the inner chest pocket of his
blazer. The shot is a cropped mid-close composition, focusing on the torso
and wallet, with the face partially visible. The background is plain light
grey, softly lit with even daylight, ensuring no harsh shadows and keeping
the focus on the subject and accessory.
- prompt: "w@llet, Close-up shot of a man\u2019s right hand holding a brown\
\ textured leather wallet with a crocodile pattern, positioned just above\
\ the back pocket of blue denim jeans. His beige t-shirt is partially visible,\
\ creating a neutral backdrop. The wallet is angled slightly outward, emphasizing\
\ its grain, pattern, and stitched detailing. The composition highlights\
\ the casual styling and practical placement, with soft, diffused lighting\
\ enhancing the textures of both the leather and denim for a natural, lifestyle-oriented\
\ presentation."
- prompt: w@llet,A man in a light grey suit jacket layered over a textured white
knit sweater poses in a semi-formal look, holding a brown leather wallet
in his right hand. His left hand adjusts the lapel of his jacket, giving
the impression of placing the wallet into the inner pocket. The wallet is
angled forward to showcase its grain texture and stitched edges. A sleek
wristwatch peeks out on his left wrist, complementing the polished look.
Background is a neutral soft grey with diffused studio lighting, creating
a clean and professional aesthetic while keeping the focus on the wallet.
neg: ''
seed: 42
walk_seed: true
guidance_scale: 4
sample_steps: 25
num_frames: 1
fps: 1
meta:
name: wallet-poses
version: '1.0'