delsin commited on
Commit
f4d42fd
·
1 Parent(s): aa14855

feat: upload test_512 lora model

Browse files
test_512_config/config_file.toml ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [model_arguments]
2
+ v2 = false
3
+ v_parameterization = false
4
+ pretrained_model_name_or_path = "/content/pretrained_model/mhmod.safetensors"
5
+
6
+ [additional_network_arguments]
7
+ no_metadata = false
8
+ unet_lr = 0.0001
9
+ text_encoder_lr = 5e-5
10
+ network_module = "lycoris.kohya"
11
+ network_dim = 16
12
+ network_alpha = 8
13
+ network_args = [ "conv_dim=8", "conv_alpha=1", "algo=lora",]
14
+ network_train_unet_only = false
15
+ network_train_text_encoder_only = false
16
+
17
+ [optimizer_arguments]
18
+ optimizer_type = "AdamW8bit"
19
+ learning_rate = 0.0001
20
+ max_grad_norm = 1.0
21
+ lr_scheduler = "constant_with_warmup"
22
+ lr_warmup_steps = 3
23
+
24
+ [dataset_arguments]
25
+ cache_latents = true
26
+ debug_dataset = false
27
+ vae_batch_size = 4
28
+
29
+ [training_arguments]
30
+ output_dir = "/content/LoRA/output"
31
+ output_name = "test_512"
32
+ save_precision = "fp16"
33
+ save_every_n_epochs = 1
34
+ train_batch_size = 6
35
+ max_token_length = 225
36
+ mem_eff_attn = false
37
+ xformers = true
38
+ max_train_epochs = 30
39
+ max_data_loader_n_workers = 8
40
+ persistent_data_loader_workers = true
41
+ seed = 2956698957
42
+ gradient_checkpointing = false
43
+ gradient_accumulation_steps = 1
44
+ mixed_precision = "fp16"
45
+ clip_skip = 2
46
+ logging_dir = "/content/LoRA/logs"
47
+ log_prefix = "test_512"
48
+ lowram = true
49
+
50
+ [sample_prompt_arguments]
51
+ sample_every_n_epochs = 1
52
+ sample_sampler = "euler_a"
53
+
54
+ [dreambooth_arguments]
55
+ prior_loss_weight = 1.0
56
+
57
+ [saving_arguments]
58
+ save_model_as = "safetensors"
test_512_config/dataset_config.toml ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [[datasets]]
2
+ resolution = 512
3
+ min_bucket_reso = 256
4
+ max_bucket_reso = 1024
5
+ caption_dropout_rate = 0
6
+ caption_tag_dropout_rate = 0
7
+ caption_dropout_every_n_epochs = 0
8
+ flip_aug = false
9
+ color_aug = false
10
+ [[datasets.subsets]]
11
+ image_dir = "/content/LoRA/train_data"
12
+ class_tokens = "tt-style"
13
+ num_repeats = 1
14
+
15
+
16
+ [general]
17
+ enable_bucket = true
18
+ caption_extension = ".txt"
19
+ shuffle_caption = true
20
+ keep_tokens = 0
21
+ bucket_reso_steps = 64
22
+ bucket_no_upscale = false
test_512_config/sample_prompt.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+
2
+ takeuchi style, masterpiece, best quality, 1girl, aqua eyes, baseball cap, blonde hair, closed mouth, earrings, green background, hat, hoop earrings, jewelry, looking at viewer, shirt, short hair, simple background, solo, upper body, yellow shirt --n lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry --w 512 --h 768 --l 7 --s 28