Glac1er commited on
Commit
2af64e3
1 Parent(s): 685e710

20K steps, 675 images, 30 epochs

Browse files
last_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/realisticVisionV20_v20.safetensors"
5
+
6
+ [optimizer_arguments]
7
+ optimizer_type = "AdamW8bit"
8
+ learning_rate = 2e-6
9
+ max_grad_norm = 1.0
10
+ train_text_encoder = false
11
+ lr_scheduler = "constant"
12
+ lr_warmup_steps = 0
13
+
14
+ [dataset_arguments]
15
+ debug_dataset = false
16
+ in_json = "/content/fine_tune/meta_lat.json"
17
+ train_data_dir = "/content/fine_tune/train_data"
18
+ dataset_repeats = 1
19
+ shuffle_caption = true
20
+ keep_tokens = 0
21
+ resolution = "512,512"
22
+ caption_dropout_rate = 0
23
+ caption_tag_dropout_rate = 0
24
+ caption_dropout_every_n_epochs = 0
25
+ color_aug = false
26
+ token_warmup_min = 1
27
+ token_warmup_step = 0
28
+
29
+ [training_arguments]
30
+ output_dir = "/content/fine_tune/output"
31
+ output_name = "last"
32
+ save_precision = "fp16"
33
+ save_n_epoch_ratio = 1
34
+ save_state = false
35
+ train_batch_size = 4
36
+ max_token_length = 225
37
+ mem_eff_attn = false
38
+ xformers = true
39
+ max_train_steps = 5000
40
+ max_data_loader_n_workers = 8
41
+ persistent_data_loader_workers = true
42
+ gradient_checkpointing = false
43
+ gradient_accumulation_steps = 1
44
+ mixed_precision = "fp16"
45
+ clip_skip = 2
46
+ logging_dir = "/content/fine_tune/logs"
47
+ log_prefix = "last"
48
+
49
+ [sample_prompt_arguments]
50
+ sample_every_n_steps = 999999
51
+ sample_sampler = "ddim"
52
+
53
+ [saving_arguments]
54
+ save_model_as = "ckpt"
last_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