GeneralGost commited on
Commit
6a6d5d6
1 Parent(s): c02b662

feat: upload archi-screencap-nai-style lora model

Browse files
archi-screencap-nai-style_config/config_file.toml ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [sdxl_arguments]
2
+ cache_text_encoder_outputs = false
3
+ no_half_vae = true
4
+ min_timestep = 0
5
+ max_timestep = 1000
6
+ shuffle_caption = true
7
+ lowram = true
8
+
9
+ [model_arguments]
10
+ pretrained_model_name_or_path = "kitty7779/ponyDiffusionV6XL"
11
+ vae = "/content/vae/sdxl_vae.safetensors"
12
+
13
+ [dataset_arguments]
14
+ debug_dataset = false
15
+ in_json = "/content/LoRA/meta_lat.json"
16
+ train_data_dir = "/content/LoRA/train_data"
17
+ dataset_repeats = 5
18
+ keep_tokens = 1
19
+ resolution = "1024,1024"
20
+ color_aug = false
21
+ token_warmup_min = 1
22
+ token_warmup_step = 0
23
+
24
+ [training_arguments]
25
+ output_dir = "/content/LoRA/output/archi-screencap-nai-style"
26
+ output_name = "archi-screencap-nai-style"
27
+ save_precision = "fp16"
28
+ save_every_n_epochs = 1
29
+ train_batch_size = 5
30
+ max_token_length = 225
31
+ mem_eff_attn = false
32
+ sdpa = true
33
+ xformers = false
34
+ max_train_epochs = 10
35
+ max_data_loader_n_workers = 8
36
+ persistent_data_loader_workers = true
37
+ seed = 23
38
+ gradient_checkpointing = true
39
+ gradient_accumulation_steps = 1
40
+ mixed_precision = "fp16"
41
+
42
+ [logging_arguments]
43
+ log_with = "tensorboard"
44
+ logging_dir = "/content/LoRA/logs"
45
+ log_prefix = "archi-screencap-nai-style"
46
+
47
+ [sample_prompt_arguments]
48
+ sample_every_n_epochs = 1
49
+ sample_sampler = "euler_a"
50
+
51
+ [saving_arguments]
52
+ save_model_as = "safetensors"
53
+
54
+ [optimizer_arguments]
55
+ optimizer_type = "AdaFactor"
56
+ learning_rate = 0.0001
57
+ max_grad_norm = 0
58
+ optimizer_args = [ "scale_parameter=False", "relative_step=False", "warmup_init=False",]
59
+ lr_scheduler = "cosine"
60
+ lr_warmup_steps = 100
61
+
62
+ [additional_network_arguments]
63
+ no_metadata = false
64
+ network_module = "networks.lora"
65
+ network_dim = 32
66
+ network_alpha = 16
67
+ network_args = []
68
+ network_train_unet_only = true
69
+
70
+ [advanced_training_config]
71
+ save_state = false
72
+ save_last_n_epochs_state = false
73
+ caption_dropout_rate = 0
74
+ caption_tag_dropout_rate = 0.5
75
+ caption_dropout_every_n_epochs = 0
76
+ min_snr_gamma = 5
archi-screencap-nai-style_config/sample_prompt.toml ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ [prompt]
2
+ negative_prompt = "3d render, smooth, plastic, blurry, grainy, low-resolution, deep-fried, oversaturated"
3
+ width = 1024
4
+ height = 1024
5
+ scale = 12
6
+ sample_steps = 28
7
+ subset = []