Joe02 commited on
Commit
5f12216
·
1 Parent(s): 9c3e675

feat: upload BLADE_style lora model

Browse files
BLADE_style_config/config_file.toml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [model_arguments]
2
+ v2 = false
3
+ v_parameterization = false
4
+ pretrained_model_name_or_path = "/content/pretrained_model/Anything-v4-5.ckpt"
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 = 32
12
+ network_alpha = 16
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"
21
+ lr_warmup_steps = 0
22
+
23
+ [dataset_arguments]
24
+ debug_dataset = false
25
+
26
+ [training_arguments]
27
+ output_dir = "/content/LoRA/output"
28
+ output_name = "BLADE_style"
29
+ save_precision = "fp16"
30
+ save_every_n_epochs = 1
31
+ train_batch_size = 6
32
+ max_token_length = 225
33
+ mem_eff_attn = false
34
+ xformers = true
35
+ max_train_epochs = 25
36
+ max_data_loader_n_workers = 8
37
+ persistent_data_loader_workers = true
38
+ gradient_checkpointing = false
39
+ gradient_accumulation_steps = 1
40
+ mixed_precision = "fp16"
41
+ clip_skip = 2
42
+ logging_dir = "/content/LoRA/logs"
43
+ log_prefix = "BLADE_style"
44
+ lowram = true
45
+
46
+ [sample_prompt_arguments]
47
+ sample_every_n_epochs = 999999
48
+ sample_sampler = "ddim"
49
+
50
+ [saving_arguments]
51
+ save_model_as = "safetensors"
BLADE_style_config/dataset_config.toml ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [[datasets]]
2
+ resolution = 512
3
+ caption_dropout_rate = 0
4
+ caption_tag_dropout_rate = 0
5
+ caption_dropout_every_n_epochs = 0
6
+ color_aug = false
7
+ [[datasets.subsets]]
8
+ image_dir = "/content/LoRA/train_data"
9
+ num_repeats = 10
10
+ metadata_file = "/content/LoRA/meta_lat.json"
11
+
12
+
13
+ [general]
14
+ shuffle_caption = true
15
+ keep_tokens = 0
BLADE_style_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