Commit
·
d7283f3
1
Parent(s):
ff06c5f
feat: upload DP-Lora lora model
Browse files
DP-Lora_config/config_file.toml
ADDED
@@ -0,0 +1,56 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
[model_arguments]
|
2 |
+
v2 = false
|
3 |
+
v_parameterization = false
|
4 |
+
pretrained_model_name_or_path = "/content/pretrained_model/epicrealism_naturalSinRC1VAE.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 |
+
optimizer_type = "AdamW8bit"
|
18 |
+
learning_rate = 0.0001
|
19 |
+
max_grad_norm = 1.0
|
20 |
+
lr_scheduler = "constant"
|
21 |
+
lr_warmup_steps = 0
|
22 |
+
|
23 |
+
[dataset_arguments]
|
24 |
+
cache_latents = true
|
25 |
+
debug_dataset = false
|
26 |
+
vae_batch_size = 4
|
27 |
+
|
28 |
+
[training_arguments]
|
29 |
+
output_dir = "/content/drive/MyDrive/LoRA/output"
|
30 |
+
output_name = "DP-Lora"
|
31 |
+
save_precision = "fp16"
|
32 |
+
save_every_n_epochs = 1
|
33 |
+
train_batch_size = 3
|
34 |
+
max_token_length = 225
|
35 |
+
mem_eff_attn = false
|
36 |
+
xformers = true
|
37 |
+
max_train_epochs = 10
|
38 |
+
max_data_loader_n_workers = 8
|
39 |
+
persistent_data_loader_workers = true
|
40 |
+
gradient_checkpointing = false
|
41 |
+
gradient_accumulation_steps = 1
|
42 |
+
mixed_precision = "fp16"
|
43 |
+
clip_skip = 2
|
44 |
+
logging_dir = "/content/LoRA/logs"
|
45 |
+
log_prefix = "DP-Lora"
|
46 |
+
lowram = false
|
47 |
+
|
48 |
+
[sample_prompt_arguments]
|
49 |
+
sample_every_n_epochs = 1
|
50 |
+
sample_sampler = "k_dpm_2"
|
51 |
+
|
52 |
+
[dreambooth_arguments]
|
53 |
+
prior_loss_weight = 1.0
|
54 |
+
|
55 |
+
[saving_arguments]
|
56 |
+
save_model_as = "safetensors"
|
DP-Lora_config/dataset_config.toml
ADDED
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
[[datasets]]
|
2 |
+
resolution = 768
|
3 |
+
min_bucket_reso = 320
|
4 |
+
max_bucket_reso = 1280
|
5 |
+
caption_dropout_rate = 0
|
6 |
+
caption_tag_dropout_rate = 0
|
7 |
+
caption_dropout_every_n_epochs = 0
|
8 |
+
flip_aug = false
|
9 |
+
color_aug = false
|
10 |
+
[[datasets.subsets]]
|
11 |
+
image_dir = "/content/LoRA/train_data"
|
12 |
+
class_tokens = "dheny"
|
13 |
+
num_repeats = 10
|
14 |
+
|
15 |
+
|
16 |
+
[general]
|
17 |
+
enable_bucket = true
|
18 |
+
caption_extension = ".txt"
|
19 |
+
shuffle_caption = true
|
20 |
+
keep_tokens = 0
|
21 |
+
bucket_reso_steps = 64
|
22 |
+
bucket_no_upscale = false
|
DP-Lora_config/sample_prompt.txt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
|
2 |
+
1boy, bokeh, half body shot, cinematic still film shot of young man at the end of the dystopian world, best quality, solo, looking at viewer, portrait, dreamlike, half body shot, photography, photorealism, photorealistic, soft focus, volumetric light
|
3 |
+
--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
|