sdasdadas commited on
Commit
f3e452d
·
1 Parent(s): ab7098a

feat: upload tesV1 lora model

Browse files
tesV1_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/AnyLoRA.safetensors"
5
+
6
+ [additional_network_arguments]
7
+ no_metadata = false
8
+ unet_lr = 0.0001
9
+ text_encoder_lr = 5e-5
10
+ network_module = "networks.lora"
11
+ network_dim = 8
12
+ network_alpha = 1
13
+ network_args = [ "conv_dim=16", "conv_alpha=8",]
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"
22
+ lr_warmup_steps = 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 = "tesV1"
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 = 50
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 = "tesV1"
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"
tesV1_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
tesV1_config/sample_prompt.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+
2
+ asher head style, solo, looking at viewer, smile, short hair, simple background, shirt, black hair, long sleeves, 1boy, holding, standing, jacket, full body, white shirt, male focus, necktie, shoes, collared shirt, belt, pants, dark skin, bag, black footwear, black eyes, formal, dark-skinned male, suit, black background, grey jacket, holding bag, grey pants, yellow necktie, flat color --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