kiluade commited on
Commit
64c64b1
·
1 Parent(s): d776996

feat: upload last lora model

Browse files
last_config/config_file.toml ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [model_arguments]
2
+ v2 = false
3
+ v_parameterization = false
4
+ pretrained_model_name_or_path = "/content/pretrained_model/Anything-v3-1.safetensors"
5
+
6
+ [additional_network_arguments]
7
+ no_metadata = false
8
+ unet_lr = 0.0001
9
+ text_encoder_lr = 4e-5
10
+ network_module = "networks.lora"
11
+ network_dim = 128
12
+ network_alpha = 64
13
+ network_train_unet_only = false
14
+ network_train_text_encoder_only = false
15
+
16
+ [optimizer_arguments]
17
+ optimizer_type = "AdamW8bit"
18
+ learning_rate = 0.0001
19
+ max_grad_norm = 1.0
20
+ lr_scheduler = "cosine_with_restarts"
21
+ lr_warmup_steps = 0
22
+ lr_scheduler_num_cycles = 0
23
+
24
+ [dataset_arguments]
25
+ cache_latents = true
26
+ debug_dataset = false
27
+
28
+ [training_arguments]
29
+ output_dir = "/content/LoRA/output"
30
+ output_name = "last"
31
+ save_precision = "fp16"
32
+ save_every_n_epochs = 1
33
+ train_batch_size = 6
34
+ max_token_length = 225
35
+ mem_eff_attn = false
36
+ xformers = true
37
+ max_train_epochs = 25
38
+ max_data_loader_n_workers = 8
39
+ persistent_data_loader_workers = true
40
+ gradient_checkpointing = false
41
+ gradient_accumulation_steps = 1
42
+ mixed_precision = "fp16"
43
+ clip_skip = 2
44
+ logging_dir = "/content/LoRA/logs"
45
+ log_prefix = "last"
46
+ lowram = true
47
+
48
+ [sample_prompt_arguments]
49
+ sample_every_n_epochs = 1
50
+ sample_sampler = "ddim"
51
+
52
+ [dreambooth_arguments]
53
+ prior_loss_weight = 1.0
54
+
55
+ [saving_arguments]
56
+ save_model_as = "safetensors"
last_config/dataset_config.toml ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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 = "mksks style"
13
+ num_repeats = 10
14
+
15
+ [[datasets.subsets]]
16
+ is_reg = true
17
+ image_dir = "/content/LoRA/reg_data"
18
+ class_tokens = "style"
19
+ num_repeats = 1
20
+
21
+
22
+ [general]
23
+ enable_bucket = true
24
+ caption_extension = ".txt"
25
+ shuffle_caption = true
26
+ keep_tokens = 0
27
+ bucket_reso_steps = 64
28
+ bucket_no_upscale = false
last_config/sample_prompt.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+
2
+ 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