Spaces:
Running
Running
| 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 | |