1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162 |
- [model_arguments]
- v2 = false
- v_parameterization = false
- pretrained_model_name_or_path = "/home/neozhang/work/resource/models/v1-5-pruned-emaonly.ckpt"
- [dataset_arguments]
- train_data_dir = "/home/neozhang/work/resource/images/swy_512"
- reg_data_dir = ""
- resolution = "512,512"
- prior_loss_weight = 1
- [additional_network_arguments]
- network_dim = 128
- network_alpha = 64
- network_train_unet_only = false
- network_train_text_encoder_only = false
- network_module = "networks.lora"
- network_args = []
- [optimizer_arguments]
- unet_lr = 1e-4
- text_encoder_lr = 1e-5
- optimizer_type = "AdamW8bit"
- lr_scheduler = "cosine_with_restarts"
- lr_warmup_steps = 0
- lr_restart_cycles = 1
- [training_arguments]
- batch_size = 4
- noise_offset = 0.0
- keep_tokens = 2
- min_bucket_reso = 256
- max_bucket_reso = 1024
- caption_extension = ".txt"
- max_token_length = 225
- seed = 1337
- xformers = true
- lowram = false
- max_train_epochs = 10
- resolution = "512,512"
- clip_skip = 2
- mixed_precision = "fp16"
- [sample_prompt_arguments]
- sample_sampler = "euler_a"
- sample_every_n_epochs = 5
- [saving_arguments]
- output_name = "swy"
- save_every_n_epochs = 1
- save_state = false
- save_model_as = "safetensors"
- output_dir = "/home/neozhang/work/train/output/zyl"
- logging_dir = "/home/neozhang/work/train/log/zyl"
- log_prefix = ""
- save_precision = "fp16"
- [others]
- cache_latents = true
- shuffle_caption = true
- enable_bucket = true
|