Ronaldo LoRA Upload

ronaldo_config/config_file.toml

@@ -0,0 +1,55 @@
+[model_arguments]
+v2 = true
+v_parameterization = false
+pretrained_model_name_or_path = "/content/pretrained_model/stable-diffusion-2-1-base.safetensors"
+
+[additional_network_arguments]
+no_metadata = false
+unet_lr = 0.0001
+text_encoder_lr = 5e-5
+network_module = "networks.lora"
+network_dim = 32
+network_alpha = 16
+network_train_unet_only = false
+network_train_text_encoder_only = false
+
+[optimizer_arguments]
+optimizer_type = "AdamW"
+learning_rate = 0.0001
+max_grad_norm = 1.0
+lr_scheduler = "constant"
+lr_warmup_steps = 0
+
+[dataset_arguments]
+cache_latents = true
+debug_dataset = false
+vae_batch_size = 4
+
+[training_arguments]
+output_dir = "/content/LoRA/output"
+output_name = "ronaldo"
+save_precision = "fp16"
+save_every_n_epochs = 1
+train_batch_size = 6
+max_token_length = 225
+mem_eff_attn = false
+xformers = true
+max_train_epochs = 10
+max_data_loader_n_workers = 8
+persistent_data_loader_workers = true
+gradient_checkpointing = false
+gradient_accumulation_steps = 1
+mixed_precision = "fp16"
+logging_dir = "/content/LoRA/logs"
+log_prefix = "ronaldo"
+lowram = true
+
+[sample_prompt_arguments]
+sample_every_n_epochs = 1
+sample_sampler = "ddim"
+
+[dreambooth_arguments]
+prior_loss_weight = 1.0
+
+[saving_arguments]
+save_model_as = "safetensors"

ronaldo_config/dataset_config.toml

@@ -0,0 +1,39 @@
+[[datasets]]
+resolution = 512
+min_bucket_reso = 256
+max_bucket_reso = 1024
+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/images"
+class_tokens = ""
+num_repeats = 30
+
+[[datasets.subsets]]
+image_dir = "/content/LoRA/images/30_glonald"
+class_tokens = "glonald"
+num_repeats = 30
+
+[[datasets.subsets]]
+image_dir = "/content/LoRA/images"
+class_tokens = ""
+num_repeats = 30
+is_reg = true
+
+[[datasets.subsets]]
+image_dir = "/content/LoRA/images/30_glonald"
+class_tokens = "glonald"
+num_repeats = 30
+is_reg = true
+
+
+[general]
+enable_bucket = true
+caption_extension = ".txt"
+shuffle_caption = true
+keep_tokens = 0
+bucket_reso_steps = 64
+bucket_no_upscale = false

ronaldo_config/sample_prompt.txt

@@ -0,0 +1,2 @@
+
+a man named glonald, playing soccer, smiling