Skip to content

Instantly share code, notes, and snippets.

@Jotschi
Created July 30, 2023 22:38
Show Gist options
  • Save Jotschi/2c7ab0183e53bc3a5f510733e1aa9f51 to your computer and use it in GitHub Desktop.
Save Jotschi/2c7ab0183e53bc3a5f510733e1aa9f51 to your computer and use it in GitHub Desktop.
TVA LyCORIS/LoKr config for kohya_ss
{
"LoRA_type": "LyCORIS/LoKr",
"adaptive_noise_scale": 0,
"additional_parameters": "",
"block_alphas": "",
"block_dims": "",
"block_lr_zero_threshold": "",
"bucket_no_upscale": true,
"bucket_reso_steps": 64,
"cache_latents": true,
"cache_latents_to_disk": true,
"caption_dropout_every_n_epochs": 0.0,
"caption_dropout_rate": 0,
"caption_extension": ".txt",
"clip_skip": "1",
"color_aug": false,
"conv_alpha": 64,
"conv_block_alphas": "",
"conv_block_dims": "",
"conv_dim": 64,
"decompose_both": false,
"dim_from_weights": false,
"down_lr_weight": "",
"enable_bucket": true,
"epoch": 40,
"factor": -1,
"flip_aug": false,
"full_bf16": false,
"full_fp16": false,
"gradient_accumulation_steps": 1.0,
"gradient_checkpointing": true,
"keep_tokens": "0",
"learning_rate": 1.0,
"logging_dir": "/extra/LoraTraining/TVA/log",
"lora_network_weights": "",
"lr_scheduler": "constant",
"lr_scheduler_num_cycles": "",
"lr_scheduler_power": "",
"lr_warmup": 0,
"max_bucket_reso": 2048,
"max_data_loader_n_workers": "0",
"max_resolution": "1024,1024",
"max_timestep": 1000,
"max_token_length": "75",
"max_train_epochs": "",
"mem_eff_attn": false,
"mid_lr_weight": "",
"min_bucket_reso": 256,
"min_snr_gamma": 10,
"min_timestep": 0,
"mixed_precision": "bf16",
"model_list": "custom",
"module_dropout": 0.1,
"multires_noise_discount": 0.2,
"multires_noise_iterations": 8,
"network_alpha": 64,
"network_dim": 64,
"network_dropout": 0,
"no_token_padding": false,
"noise_offset": 0.0357,
"noise_offset_type": "Multires",
"num_cpu_threads_per_process": 2,
"optimizer": "Prodigy",
"optimizer_args": "",
"output_dir": "/extra/LoraTraining/TVA/model",
"output_name": "tva",
"persistent_data_loader_workers": false,
"pretrained_model_name_or_path": "/extra/ComfyUI/models/checkpoints/sd_xl_base_1.0.safetensors",
"prior_loss_weight": 1.0,
"random_crop": false,
"rank_dropout": 0.1,
"reg_data_dir": "",
"resume": "",
"sample_every_n_epochs": 0,
"sample_every_n_steps": 0,
"sample_prompts": "",
"sample_sampler": "euler_a",
"save_every_n_epochs": 1,
"save_every_n_steps": 450,
"save_last_n_steps": 0,
"save_last_n_steps_state": 0,
"save_model_as": "safetensors",
"save_precision": "fp16",
"save_state": false,
"scale_v_pred_loss_like_noise_pred": false,
"scale_weight_norms": 0,
"sdxl": true,
"sdxl_cache_text_encoder_outputs": false,
"sdxl_no_half_vae": true,
"seed": "12345",
"shuffle_caption": false,
"stop_text_encoder_training": 0,
"text_encoder_lr": 1.0,
"train_batch_size": 8,
"train_data_dir": "/extra/LoraTraining/TVA/images",
"train_on_input": true,
"training_comment": "",
"unet_lr": 1.0,
"unit": 1,
"up_lr_weight": "",
"use_cp": true,
"use_wandb": false,
"v2": false,
"v_parameterization": false,
"vae_batch_size": 0,
"wandb_api_key": "",
"weighted_captions": false,
"xformers": true
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment