feat: upload yami lora model

yami_config/config_file.toml

@@ -23,6 +23,7 @@ lr_warmup_steps = 0
 lr_scheduler_num_cycles = 2
 
 [dataset_arguments]
+cache_latents = true
 debug_dataset = false
 
 [training_arguments]
@@ -49,5 +50,8 @@ lowram = true
 sample_every_n_epochs = 999999
 sample_sampler = "ddim"
 
+[dreambooth_arguments]
+prior_loss_weight = 1.0
+
 [saving_arguments]
 save_model_as = "safetensors"

yami_config/dataset_config.toml

@@ -1,15 +1,28 @@
 [[datasets]]
 resolution = 768
+min_bucket_reso = 320
+max_bucket_reso = 1280
 caption_dropout_rate = 0
 caption_tag_dropout_rate = 0
 caption_dropout_every_n_epochs = 0
+flip_aug = false
 color_aug = false
 [[datasets.subsets]]
 image_dir = "/content/LoRA/train_data"
+class_tokens = "yami "
 num_repeats = 3
-metadata_file = "/content/LoRA/meta_lat.json"
+
+[[datasets.subsets]]
+is_reg = true
+image_dir = "/content/LoRA/reg_data"
+class_tokens = ""
+num_repeats = 1
 
 
 [general]
+enable_bucket = true
+caption_extension = ".txt"
 shuffle_caption = true
 keep_tokens = 1
+bucket_reso_steps = 64
+bucket_no_upscale = false