Gary0076 commited on
Commit
3728c10
·
1 Parent(s): 23a0d17

feat: upload sutegoro-shiina lora model

Browse files
sutegoro-shiina_config/config_file.toml ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [model_arguments]
2
+ v2 = false
3
+ v_parameterization = false
4
+ pretrained_model_name_or_path = "/content/pretrained_model/Animefull-final-pruned.ckpt"
5
+
6
+ [additional_network_arguments]
7
+ no_metadata = false
8
+ unet_lr = 0.00016
9
+ text_encoder_lr = 3e-5
10
+ network_module = "networks.lora"
11
+ network_dim = 128
12
+ network_alpha = 128
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.00016
19
+ max_grad_norm = 1.0
20
+ lr_scheduler = "cosine_with_restarts"
21
+ lr_warmup_steps = 0
22
+ lr_scheduler_num_cycles = 3
23
+
24
+ [dataset_arguments]
25
+ debug_dataset = false
26
+
27
+ [training_arguments]
28
+ output_dir = "/content/LoRA/output"
29
+ output_name = "sutegoro-shiina"
30
+ save_precision = "fp16"
31
+ save_every_n_epochs = 25
32
+ train_batch_size = 3
33
+ max_token_length = 225
34
+ mem_eff_attn = false
35
+ xformers = true
36
+ max_train_epochs = 5
37
+ max_data_loader_n_workers = 8
38
+ persistent_data_loader_workers = true
39
+ gradient_checkpointing = false
40
+ gradient_accumulation_steps = 1
41
+ mixed_precision = "fp16"
42
+ clip_skip = 2
43
+ logging_dir = "/content/LoRA/logs"
44
+ log_prefix = "sutegoro-shiina"
45
+ lowram = true
46
+
47
+ [sample_prompt_arguments]
48
+ sample_every_n_epochs = 999999
49
+ sample_sampler = "ddim"
50
+
51
+ [saving_arguments]
52
+ save_model_as = "safetensors"
sutegoro-shiina_config/dataset_config.toml ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [[datasets]]
2
+ resolution = 768
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 = 30
10
+ metadata_file = "/content/LoRA/meta_lat.json"
11
+
12
+
13
+ [general]
14
+ shuffle_caption = true
15
+ keep_tokens = 0
sutegoro-shiina_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