Squirz commited on
Commit
91b5378
1 Parent(s): 1da2417

feat: upload Rosa2 lora model

Browse files
Rosa2_config/config_file.toml ADDED
@@ -0,0 +1,66 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [model_arguments]
2
+ v2 = false
3
+ v_parameterization = false
4
+ pretrained_model_name_or_path = "/content/pretrained_model/meinamix_meinaV9.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 = 32
12
+ network_alpha = 16
13
+ network_train_unet_only = false
14
+ network_train_text_encoder_only = false
15
+
16
+ [optimizer_arguments]
17
+ min_snr_gamma = 4
18
+ optimizer_type = "AdamW8bit"
19
+ learning_rate = 0.0001
20
+ max_grad_norm = 1.0
21
+ lr_scheduler = "cosine_with_restarts"
22
+ lr_warmup_steps = 0.05
23
+ lr_scheduler_num_cycles = 0
24
+
25
+ [dataset_arguments]
26
+ debug_dataset = false
27
+ in_json = "/content/LoRA/meta_lat.json"
28
+ train_data_dir = "/content/LoRA/train_data/rosa"
29
+ dataset_repeats = 10
30
+ shuffle_caption = true
31
+ keep_tokens = 1
32
+ resolution = "512,512"
33
+ caption_dropout_rate = 0
34
+ caption_tag_dropout_rate = 0
35
+ caption_dropout_every_n_epochs = 0
36
+ color_aug = false
37
+ token_warmup_min = 1
38
+ token_warmup_step = 0
39
+
40
+ [training_arguments]
41
+ output_dir = "/content/LoRA/output"
42
+ output_name = "Rosa2"
43
+ save_precision = "fp16"
44
+ save_n_epoch_ratio = 1
45
+ train_batch_size = 2
46
+ max_token_length = 225
47
+ mem_eff_attn = false
48
+ xformers = true
49
+ max_train_epochs = 10
50
+ max_data_loader_n_workers = 8
51
+ persistent_data_loader_workers = true
52
+ gradient_checkpointing = true
53
+ gradient_accumulation_steps = 2
54
+ mixed_precision = "fp16"
55
+ clip_skip = 2
56
+ logging_dir = "/content/LoRA/logs"
57
+ log_prefix = "Rosa2"
58
+ noise_offset = 0.05
59
+ lowram = true
60
+
61
+ [sample_prompt_arguments]
62
+ sample_every_n_epochs = 1
63
+ sample_sampler = "euler"
64
+
65
+ [saving_arguments]
66
+ save_model_as = "safetensors"
Rosa2_config/sample_prompt.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+
2
+ masterpiece, best quality, 1girl, solo, long hair, looking at viewer, open mouth, bangs, brown hair, brown eyes, blunt bangs, black background, hime cut, rosa, long 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