sdasdadas commited on
Commit
403f069
·
1 Parent(s): f8e37b4

feat: upload lora_rackhamV2 lora model

Browse files
lora_rackhamV2_config/config_file.toml ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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 = "networks.lora"
11
+ network_dim = 8
12
+ network_alpha = 1
13
+ network_train_unet_only = false
14
+ network_train_text_encoder_only = false
15
+
16
+ [optimizer_arguments]
17
+ min_snr_gamma = 5.0
18
+ optimizer_type = "AdamW8bit"
19
+ learning_rate = 0.0001
20
+ max_grad_norm = 1.0
21
+ lr_scheduler = "constant_with_warmup"
22
+ lr_warmup_steps = 135
23
+
24
+ [dataset_arguments]
25
+ cache_latents = true
26
+ debug_dataset = false
27
+ vae_batch_size = 4
28
+
29
+ [training_arguments]
30
+ output_dir = "/content/LoRA/output"
31
+ output_name = "lora_rackhamV2"
32
+ save_precision = "fp16"
33
+ save_every_n_epochs = 1
34
+ train_batch_size = 2
35
+ max_token_length = 225
36
+ mem_eff_attn = false
37
+ xformers = true
38
+ max_train_epochs = 6
39
+ max_data_loader_n_workers = 8
40
+ persistent_data_loader_workers = true
41
+ seed = 46
42
+ gradient_checkpointing = false
43
+ gradient_accumulation_steps = 1
44
+ mixed_precision = "fp16"
45
+ clip_skip = 2
46
+ logging_dir = "/content/LoRA/logs"
47
+ log_prefix = "lora_rackhamV2"
48
+ noise_offset = 0.03
49
+ lowram = true
50
+
51
+ [sample_prompt_arguments]
52
+ sample_every_n_epochs = 1
53
+ sample_sampler = "ddim"
54
+
55
+ [dreambooth_arguments]
56
+ prior_loss_weight = 1.0
57
+
58
+ [saving_arguments]
59
+ save_model_as = "safetensors"
lora_rackhamV2_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
lora_rackhamV2_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