claoire commited on
Commit
dcc5bc3
1 Parent(s): c1f19b6

feat: upload ckft lora model

Browse files
ckft_config/config_file.toml ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [model_arguments]
2
+ v2 = false
3
+ v_parameterization = false
4
+ pretrained_model_name_or_path = "/content/pretrained_model/epicrealism_naturalSin.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 = 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 = "constant"
21
+ lr_warmup_steps = 0
22
+
23
+ [dataset_arguments]
24
+ debug_dataset = false
25
+ in_json = "/content/LoRA/meta_lat.json"
26
+ train_data_dir = "/content/LoRA/train_data"
27
+ dataset_repeats = 1
28
+ shuffle_caption = true
29
+ keep_tokens = 0
30
+ resolution = "512,512"
31
+ caption_dropout_rate = 0
32
+ caption_tag_dropout_rate = 0
33
+ caption_dropout_every_n_epochs = 0
34
+ color_aug = false
35
+ token_warmup_min = 1
36
+ token_warmup_step = 0
37
+
38
+ [training_arguments]
39
+ output_dir = "/content/LoRA/output"
40
+ output_name = "ckft"
41
+ save_precision = "fp16"
42
+ save_every_n_epochs = 1
43
+ train_batch_size = 6
44
+ max_token_length = 225
45
+ mem_eff_attn = false
46
+ xformers = true
47
+ max_train_epochs = 10
48
+ max_data_loader_n_workers = 8
49
+ persistent_data_loader_workers = true
50
+ gradient_checkpointing = false
51
+ gradient_accumulation_steps = 1
52
+ mixed_precision = "fp16"
53
+ clip_skip = 2
54
+ logging_dir = "/content/LoRA/logs"
55
+ log_prefix = "ckft"
56
+ lowram = true
57
+
58
+ [sample_prompt_arguments]
59
+ sample_every_n_epochs = 1
60
+ sample_sampler = "ddim"
61
+
62
+ [saving_arguments]
63
+ save_model_as = "safetensors"
ckft_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