01_Studio_v1 / config.yaml
Sri2901's picture
Upload model via CLI
b3f42f5 verified
job: extension
config:
name: 01_Studio_v1
process:
- type: ui_trainer
training_folder: /workspace/ai-toolkit/output
sqlite_db_path: /workspace/ai-toolkit/aitk_db.db
device: cuda
trigger_word: studi@
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: 500
max_step_saves_to_keep: 7
save_format: diffusers
push_to_hub: false
datasets:
- folder_path: /workspace/ai-toolkit/datasets/01_studio_v1
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: []
train:
batch_size: 1
bypass_guidance_embedding: false
steps: 4000
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
lr: 0.0004
ema_config:
use_ema: false
ema_decay: 0.99
dtype: bf16
diff_output_preservation: false
diff_output_preservation_multiplier: 1
diff_output_preservation_class: person
model:
name_or_path: black-forest-labs/FLUX.1-dev
quantize: false
quantize_te: true
arch: flux
low_vram: false
model_kwargs: {}
sample:
sampler: flowmatch
sample_every: 500
width: 1024
height: 1024
prompts:
- A photo-realistic shoot from a profile camera angle about a woman wearing
a blue and white floral print dress with puff sleeves and a tie waist, standing
in profile against a plain grey background. the image also shows a woman standing
in the middle of the frame, facing away from the camera, wearing a white dress
with blue floral patterns and a ruffled hem. on the right side of the image,
a woman appears to be in her mid-twenties, with long black hair styled in
loose waves, wearing earrings and a bracelet on her wrist. her expression
is neutral, and she is standing with her arms at her sides.
- A woman standing confidently in the middle of the image, wearing a black dress
with white piping and a blue handbag. She has short black hair styled in a
sleek manner and is wearing gold earrings. her posture is confident and strong,
and she is standing with one hand on her hip and the other resting on her
thigh. the background is a simple gradient of light blue to white, creating
a minimalistic look. the lighting is soft and even, highlighting her features
and creating a sense of sophistication. the image is shot from a full body
angle, focusing on the woman's full body and the handbag in her left hand.
- A woman standing confidently in the middle of a plain grey background, wearing
a blue and white floral print dress with puff sleeves and black ankle boots.
she is holding a small black clutch in her left hand and wearing black sunglasses.
her long black hair cascades down her back, framing her face perfectly. the
dress has a gathered waistline and a flowy skirt that falls just above her
knees. the woman is standing with her arms at her sides, looking directly
at the camera with a neutral expression. her black boots are ankle-length
and reach up to her knees, adding a touch of sophistication to her outfit.
the background is simple and minimalistic, allowing the focus to remain on
the woman's figure and the subtle pattern of her dress.
neg: ''
seed: 42
walk_seed: true
guidance_scale: 3.5
sample_steps: 30
num_frames: 1
fps: 1
meta:
name: 01_Studio_v1
version: '1.0'