Upload config.toml with huggingface_hub

config.toml

@@ -0,0 +1,55 @@
+bucket_no_upscale = true
+bucket_reso_steps = 64
+cache_latents = true
+caption_extension = ".txt"
+clip_skip = 1
+dynamo_backend = "no"
+enable_bucket = true
+epoch = 20
+gradient_accumulation_steps = 1
+huber_c = 0.1
+huber_schedule = "snr"
+huggingface_repo_id = "BKM1804/lora_cluster"
+huggingface_repo_type = "model"
+huggingface_repo_visibility = "public"
+learning_rate = 0.0001
+loss_type = "l2"
+lr_scheduler = "cosine_with_restarts"
+lr_scheduler_args = []
+lr_scheduler_num_cycles = 1
+lr_scheduler_power = 1
+lr_warmup_steps = 300
+max_bucket_reso = 2048
+max_data_loader_n_workers = 0
+max_grad_norm = 1
+max_timestep = 1000
+max_token_length = 75
+max_train_steps = 3000
+min_bucket_reso = 256
+min_snr_gamma = 5
+mixed_precision = "fp16"
+multires_noise_discount = 0.3
+network_alpha = 128
+network_args = []
+network_dim = 128
+network_module = "networks.lora"
+no_half_vae = true
+noise_offset = 0.1
+noise_offset_type = "Original"
+optimizer_args = []
+optimizer_type = "AdamW"
+output_dir = "/workspace/kohya_ss/outputs"
+output_name = "last"
+pretrained_model_name_or_path = "stabilityai/stable-diffusion-xl-base-1.0"
+prior_loss_weight = 1
+resolution = "512"
+sample_prompts = "/workspace/kohya_ss/outputs/prompt.txt"
+sample_sampler = "euler_a"
+save_every_n_epochs = 1
+save_model_as = "safetensors"
+save_precision = "fp16"
+text_encoder_lr = 4e-5
+train_batch_size = 1
+train_data_dir = "/workspace/data/img"
+unet_lr = 0.0001
+xformers = true