x1101 commited on
Commit
282af3a
1 Parent(s): cdd512b
Aurora_config/config_file.toml ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [model_arguments]
2
+ v2 = false
3
+ v_parameterization = false
4
+ pretrained_model_name_or_path = "/content/pretrained_model/anylora_novaeFp16Pruned.ckpt"
5
+ vae = "/content/vae/anime.vae.pt"
6
+
7
+ [additional_network_arguments]
8
+ no_metadata = false
9
+ unet_lr = 0.0005
10
+ text_encoder_lr = 0.0001
11
+ network_module = "lycoris.kohya"
12
+ network_dim = 16
13
+ network_alpha = 8
14
+ network_args = [ "conv_dim=16", "conv_alpha=8", "algo=lora",]
15
+ network_train_unet_only = false
16
+ network_train_text_encoder_only = false
17
+
18
+ [optimizer_arguments]
19
+ optimizer_type = "AdamW8bit"
20
+ learning_rate = 0.0005
21
+ max_grad_norm = 1.0
22
+ lr_scheduler = "cosine_with_restarts"
23
+ lr_warmup_steps = 0
24
+ lr_scheduler_num_cycles = 3
25
+
26
+ [dataset_arguments]
27
+ debug_dataset = false
28
+
29
+ [training_arguments]
30
+ output_dir = "/content/LoRA/output"
31
+ output_name = "Aurora"
32
+ save_precision = "fp16"
33
+ save_every_n_epochs = 3
34
+ train_batch_size = 2
35
+ max_token_length = 225
36
+ mem_eff_attn = false
37
+ xformers = true
38
+ max_train_epochs = 10
39
+ max_data_loader_n_workers = 8
40
+ persistent_data_loader_workers = true
41
+ gradient_checkpointing = false
42
+ gradient_accumulation_steps = 1
43
+ mixed_precision = "fp16"
44
+ clip_skip = 2
45
+ logging_dir = "/content/LoRA/logs"
46
+ log_prefix = "Aurora"
47
+ lowram = true
48
+
49
+ [sample_prompt_arguments]
50
+ sample_every_n_epochs = 999999
51
+ sample_sampler = "ddim"
52
+
53
+ [saving_arguments]
54
+ save_model_as = "safetensors"
Aurora_config/dataset_config.toml ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [[datasets]]
2
+ resolution = 512
3
+ caption_dropout_rate = 0
4
+ caption_tag_dropout_rate = 0
5
+ caption_dropout_every_n_epochs = 0
6
+ color_aug = false
7
+ [[datasets.subsets]]
8
+ image_dir = "/content/LoRA/dataset"
9
+ num_repeats = 15
10
+ metadata_file = "/content/LoRA/tags/meta_lat.json"
11
+
12
+
13
+ [general]
14
+ shuffle_caption = true
15
+ keep_tokens = 0
Aurora_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