fazeelzafar
commited on
Commit
•
663cd6c
1
Parent(s):
9a70f05
feat: upload SD_Test_Project lora model
Browse files
SD_Test_Project_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/v1-5-pruned-emaonly.ckpt"
|
5 |
+
vae = "/content/vae/stablediffusion.vae.pt"
|
6 |
+
|
7 |
+
[additional_network_arguments]
|
8 |
+
no_metadata = false
|
9 |
+
unet_lr = 0.0001
|
10 |
+
text_encoder_lr = 5e-5
|
11 |
+
network_module = "networks.lora"
|
12 |
+
network_dim = 8
|
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 = "AdamW8bit"
|
20 |
+
learning_rate = 0.0001
|
21 |
+
max_grad_norm = 1.0
|
22 |
+
lr_scheduler = "constant"
|
23 |
+
lr_warmup_steps = 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"
|
29 |
+
dataset_repeats = 10
|
30 |
+
shuffle_caption = true
|
31 |
+
keep_tokens = 0
|
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/drive/MyDrive/LoRA/output"
|
42 |
+
output_name = "SD_Test_Project"
|
43 |
+
save_precision = "fp16"
|
44 |
+
save_every_n_epochs = 1
|
45 |
+
train_batch_size = 6
|
46 |
+
max_token_length = 225
|
47 |
+
mem_eff_attn = false
|
48 |
+
xformers = true
|
49 |
+
max_train_epochs = 25
|
50 |
+
max_data_loader_n_workers = 8
|
51 |
+
persistent_data_loader_workers = true
|
52 |
+
gradient_checkpointing = true
|
53 |
+
gradient_accumulation_steps = 1
|
54 |
+
mixed_precision = "fp16"
|
55 |
+
clip_skip = 2
|
56 |
+
logging_dir = "/content/LoRA/logs"
|
57 |
+
log_prefix = "SD_Test_Project"
|
58 |
+
noise_offset = 0.2
|
59 |
+
lowram = true
|
60 |
+
|
61 |
+
[sample_prompt_arguments]
|
62 |
+
sample_every_n_epochs = 1
|
63 |
+
sample_sampler = "ddim"
|
64 |
+
|
65 |
+
[saving_arguments]
|
66 |
+
save_model_as = "safetensors"
|
SD_Test_Project_config/sample_prompt.txt
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
|
2 |
+
masterpiece, best quality, suit, man in a suit, professional image, professional background, closed mouth, professional headshot, headshot with body, headshot, smiling, smile, hd skin tone, simple background, solo, formal shirt, formal attire, casual attire, DSLR camera with a high-resolution, wide-angle to capture the full context, high detailing, realistic, UHD, many details, perfect hands, highres:1.2, detailed light:1.2 --n lowres, bad anatomy, bad hands, wrong hands, wrong features, bad hands, no headshot, image without face, no fingers, no hands, poor quality background, deformed hands, deformed fingers, text, deformed faces, cartoonish faces, cartoonish objects, cropped images, no face , error, missing fingers, cropped, out of frame, random features, weird features, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry --w 512 --h 512 --l 7 --s 28
|