sdasdadas commited on
Commit
0f83bae
·
1 Parent(s): fc4231a

feat: upload loco_rackhamV1 lora model

Browse files
loco_rackhamV1_config/config_file.toml ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [model_arguments]
2
+ v2 = false
3
+ v_parameterization = false
4
+ pretrained_model_name_or_path = "/content/pretrained_model/anyloraCheckpoint_novaeFp16.safetensors"
5
+
6
+ [additional_network_arguments]
7
+ no_metadata = false
8
+ unet_lr = 0.0005
9
+ text_encoder_lr = 5e-5
10
+ network_module = "lycoris.kohya"
11
+ network_dim = 8
12
+ network_alpha = 1
13
+ network_args = [ "conv_dim=16", "conv_alpha=8", "algo=lora",]
14
+ network_train_unet_only = false
15
+ network_train_text_encoder_only = false
16
+
17
+ [optimizer_arguments]
18
+ min_snr_gamma = 5.0
19
+ optimizer_type = "AdamW8bit"
20
+ learning_rate = 0.0005
21
+ max_grad_norm = 1.0
22
+ lr_scheduler = "constant_with_warmup"
23
+ lr_warmup_steps = 135
24
+
25
+ [dataset_arguments]
26
+ cache_latents = true
27
+ debug_dataset = false
28
+ vae_batch_size = 4
29
+
30
+ [training_arguments]
31
+ output_dir = "/content/LoRA/output"
32
+ output_name = "loco_rackhamV1"
33
+ save_precision = "fp16"
34
+ save_every_n_epochs = 1
35
+ train_batch_size = 2
36
+ max_token_length = 225
37
+ mem_eff_attn = false
38
+ xformers = true
39
+ max_train_epochs = 8
40
+ max_data_loader_n_workers = 8
41
+ persistent_data_loader_workers = true
42
+ seed = 46
43
+ gradient_checkpointing = false
44
+ gradient_accumulation_steps = 1
45
+ mixed_precision = "fp16"
46
+ clip_skip = 2
47
+ logging_dir = "/content/LoRA/logs"
48
+ log_prefix = "loco_rackhamV1"
49
+ noise_offset = 0.03
50
+ lowram = true
51
+
52
+ [sample_prompt_arguments]
53
+ sample_every_n_epochs = 1
54
+ sample_sampler = "ddim"
55
+
56
+ [dreambooth_arguments]
57
+ prior_loss_weight = 1.0
58
+
59
+ [saving_arguments]
60
+ save_model_as = "safetensors"
loco_rackhamV1_config/dataset_config.toml ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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 = "rackham"
13
+ num_repeats = 5
14
+
15
+
16
+ [general]
17
+ enable_bucket = true
18
+ caption_extension = ".caption"
19
+ shuffle_caption = true
20
+ keep_tokens = 1
21
+ bucket_reso_steps = 64
22
+ bucket_no_upscale = false
loco_rackhamV1_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