delsin commited on
Commit
8770041
·
1 Parent(s): e562f04

feat: upload chocochip lora model

Browse files
chocochip_config/config_file.toml ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [model_arguments]
2
+ v2 = false
3
+ v_parameterization = false
4
+ pretrained_model_name_or_path = "/content/pretrained_model/anything-v4.5.safetensors"
5
+
6
+ [additional_network_arguments]
7
+ no_metadata = false
8
+ unet_lr = 0.0001
9
+ text_encoder_lr = 3e-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 = "constant"
21
+ lr_warmup_steps = 0
22
+
23
+ [dataset_arguments]
24
+ cache_latents = true
25
+ debug_dataset = false
26
+
27
+ [training_arguments]
28
+ output_dir = "/content/LoRA/output"
29
+ output_name = "chocochip"
30
+ save_precision = "fp16"
31
+ save_every_n_epochs = 1
32
+ train_batch_size = 6
33
+ max_token_length = 225
34
+ mem_eff_attn = false
35
+ xformers = true
36
+ max_train_epochs = 20
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 = "chocochip"
45
+ lowram = true
46
+
47
+ [sample_prompt_arguments]
48
+ sample_every_n_epochs = 1
49
+ sample_sampler = "ddim"
50
+
51
+ [dreambooth_arguments]
52
+ prior_loss_weight = 1.0
53
+
54
+ [saving_arguments]
55
+ save_model_as = "safetensors"
chocochip_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 = "chocochip 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
chocochip_config/sample_prompt.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+
2
+ chocochip, 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