sdasdadas commited on
Commit
87c5383
·
1 Parent(s): 38a0837

feat: upload productpkgV1 lora model

Browse files
productpkgV1_config/config_file.toml ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [model_arguments]
2
+ v2 = false
3
+ v_parameterization = false
4
+ pretrained_model_name_or_path = "/content/pretrained_model/revAnimated_v122.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 = 64
12
+ network_alpha = 32
13
+ network_args = []
14
+ network_train_unet_only = false
15
+ network_train_text_encoder_only = false
16
+
17
+ [optimizer_arguments]
18
+ optimizer_type = "AdamW8bit"
19
+ learning_rate = 0.0001
20
+ max_grad_norm = 1.0
21
+ optimizer_args = []
22
+ lr_scheduler = "constant"
23
+ lr_warmup_steps = 0
24
+
25
+ [dataset_arguments]
26
+ cache_latents = true
27
+ cache_latents_to_disk = true
28
+ debug_dataset = false
29
+ vae_batch_size = 4
30
+
31
+ [training_arguments]
32
+ output_dir = "/content/LoRA/output"
33
+ output_name = "productpkgV1"
34
+ save_precision = "fp16"
35
+ save_every_n_epochs = 1
36
+ train_batch_size = 2
37
+ max_token_length = 225
38
+ mem_eff_attn = false
39
+ xformers = true
40
+ max_train_epochs = 25
41
+ max_data_loader_n_workers = 8
42
+ persistent_data_loader_workers = true
43
+ seed = 46
44
+ gradient_accumulation_steps = 1
45
+ mixed_precision = "fp16"
46
+ clip_skip = 1
47
+ lowram = true
48
+
49
+ [logging_arguments]
50
+ log_with = "tensorboard"
51
+ logging_dir = "/content/LoRA/logs"
52
+ log_prefix = "productpkgV1"
53
+
54
+ [sample_prompt_arguments]
55
+ sample_every_n_epochs = 1
56
+ sample_sampler = "k_dpm_2_a"
57
+
58
+ [dreambooth_arguments]
59
+ prior_loss_weight = 1.0
60
+
61
+ [saving_arguments]
62
+ save_model_as = "safetensors"
productpkgV1_config/dataset_config.toml ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [[datasets]]
2
+ resolution = 512
3
+ min_bucket_reso = 256
4
+ max_bucket_reso = 1024
5
+ flip_aug = false
6
+ [[datasets.subsets]]
7
+ image_dir = "/content/LoRA/train_data"
8
+ num_repeats = 20
9
+
10
+ [[datasets.subsets]]
11
+ image_dir = "/content/LoRA/train_data/concept1"
12
+ num_repeats = 20
13
+
14
+ [[datasets.subsets]]
15
+ image_dir = "/content/LoRA/train_data/concept2"
16
+ num_repeats = 20
17
+
18
+
19
+ [general]
20
+ enable_bucket = true
21
+ caption_extension = ".txt"
22
+ shuffle_caption = true
23
+ keep_tokens = 1
productpkgV1_config/sample_prompt.toml ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [prompt]
2
+ negative_prompt = "(worst quality, low quality:1.4), "
3
+ width = 512
4
+ height = 768
5
+ scale = 7
6
+ sample_steps = 28
7
+ [[prompt.subset]]
8
+ prompt = "masterpiece, best quality, cncpt2,"
9
+
10
+ [[prompt.subset]]
11
+ prompt = "masterpiece, best quality, cncpt2,"
12
+