ada233 commited on
Commit
098ea9e
·
1 Parent(s): cdf72cc
alpha9_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/AnyLoRA.safetensors"
5
+ vae = "/content/vae/anime.vae.pt"
6
+
7
+ [additional_network_arguments]
8
+ no_metadata = false
9
+ unet_lr = 0.001
10
+ text_encoder_lr = 0.0005
11
+ network_module = "networks.lora"
12
+ network_dim = 32
13
+ network_alpha = 1
14
+ network_train_unet_only = false
15
+ network_train_text_encoder_only = false
16
+
17
+ [optimizer_arguments]
18
+ min_snr_gamma = 5
19
+ optimizer_type = "Lion"
20
+ learning_rate = 0.001
21
+ max_grad_norm = 1.0
22
+ lr_scheduler = "cosine_with_restarts"
23
+ lr_warmup_steps = 0
24
+ lr_scheduler_num_cycles = 0
25
+
26
+ [dataset_arguments]
27
+ cache_latents = true
28
+ debug_dataset = false
29
+ vae_batch_size = 4
30
+
31
+ [training_arguments]
32
+ output_dir = "/content/LoRA/output"
33
+ output_name = "alpha9"
34
+ save_precision = "fp16"
35
+ save_every_n_epochs = 1
36
+ train_batch_size = 6
37
+ max_token_length = 225
38
+ mem_eff_attn = false
39
+ xformers = true
40
+ max_train_epochs = 15
41
+ max_data_loader_n_workers = 8
42
+ persistent_data_loader_workers = true
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 = "alpha9"
49
+ noise_offset = 0.1
50
+ lowram = true
51
+
52
+ [sample_prompt_arguments]
53
+ sample_every_n_epochs = 999999
54
+ sample_sampler = "ddim"
55
+
56
+ [dreambooth_arguments]
57
+ prior_loss_weight = 1.0
58
+
59
+ [saving_arguments]
60
+ save_model_as = "safetensors"
alpha9_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 = true
9
+ color_aug = false
10
+ [[datasets.subsets]]
11
+ image_dir = "/content/LoRA/train_data"
12
+ class_tokens = ""
13
+ num_repeats = 15
14
+
15
+
16
+ [general]
17
+ enable_bucket = true
18
+ caption_extension = ".txt"
19
+ shuffle_caption = true
20
+ keep_tokens = 0
21
+ bucket_reso_steps = 64
22
+ bucket_no_upscale = false
alpha9_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