mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-01-26 16:39:47 +00:00
92 lines
2.8 KiB
YAML
92 lines
2.8 KiB
YAML
---
|
|
job: extension
|
|
config:
|
|
name: test_v1
|
|
process:
|
|
- type: 'textual_inversion_trainer'
|
|
training_folder: "out/TI"
|
|
device: cuda:0
|
|
# for tensorboard logging
|
|
log_dir: "out/.tensorboard"
|
|
embedding:
|
|
trigger: "your_trigger_here"
|
|
tokens: 12
|
|
init_words: "man with short brown hair"
|
|
save_format: "safetensors" # 'safetensors' or 'pt'
|
|
save:
|
|
dtype: float16 # precision to save
|
|
save_every: 100 # save every this many steps
|
|
max_step_saves_to_keep: 5 # only affects step counts
|
|
datasets:
|
|
- folder_path: "/path/to/dataset"
|
|
caption_ext: "txt"
|
|
default_caption: "[trigger]"
|
|
buckets: true
|
|
resolution: 512
|
|
train:
|
|
noise_scheduler: "ddpm" # or "ddpm", "lms", "euler_a"
|
|
steps: 3000
|
|
weight_jitter: 0.0
|
|
lr: 5e-5
|
|
train_unet: false
|
|
gradient_checkpointing: true
|
|
train_text_encoder: false
|
|
optimizer: "adamw"
|
|
# optimizer: "prodigy"
|
|
optimizer_params:
|
|
weight_decay: 1e-2
|
|
lr_scheduler: "constant"
|
|
max_denoising_steps: 1000
|
|
batch_size: 4
|
|
dtype: bf16
|
|
xformers: true
|
|
min_snr_gamma: 5.0
|
|
# skip_first_sample: true
|
|
noise_offset: 0.0 # not needed for this
|
|
model:
|
|
# objective reality v2
|
|
name_or_path: "https://civitai.com/models/128453?modelVersionId=142465"
|
|
is_v2: false # for v2 models
|
|
is_xl: false # for SDXL models
|
|
is_v_pred: false # for v-prediction models (most v2 models)
|
|
sample:
|
|
sampler: "ddpm" # must match train.noise_scheduler
|
|
sample_every: 100 # sample every this many steps
|
|
width: 512
|
|
height: 512
|
|
prompts:
|
|
- "photo of [trigger] laughing"
|
|
- "photo of [trigger] smiling"
|
|
- "[trigger] close up"
|
|
- "dark scene [trigger] frozen"
|
|
- "[trigger] nighttime"
|
|
- "a painting of [trigger]"
|
|
- "a drawing of [trigger]"
|
|
- "a cartoon of [trigger]"
|
|
- "[trigger] pixar style"
|
|
- "[trigger] costume"
|
|
neg: ""
|
|
seed: 42
|
|
walk_seed: false
|
|
guidance_scale: 7
|
|
sample_steps: 20
|
|
network_multiplier: 1.0
|
|
|
|
logging:
|
|
log_every: 10 # log every this many steps
|
|
use_wandb: false # not supported yet
|
|
verbose: false
|
|
|
|
# You can put any information you want here, and it will be saved in the model.
|
|
# The below is an example, but you can put your grocery list in it if you want.
|
|
# It is saved in the model so be aware of that. The software will include this
|
|
# plus some other information for you automatically
|
|
meta:
|
|
# [name] gets replaced with the name above
|
|
name: "[name]"
|
|
# version: '1.0'
|
|
# creator:
|
|
# name: Your Name
|
|
# email: your@gmail.com
|
|
# website: https://your.website
|