e10
Browse files- config/config_file.toml +57 -0
- config/dataset_config.toml +22 -0
- config/sample_prompt.txt +2 -0
- input/00.caption +1 -0
- input/00.jpg +0 -0
- input/01.caption +1 -0
- input/01.jpg +0 -0
- input/02.caption +1 -0
- input/02.jpg +0 -0
- input/03.caption +1 -0
- input/03.jpg +0 -0
- input/04.caption +1 -0
- input/04.jpg +0 -0
- input/05.caption +1 -0
- input/05.jpg +0 -0
- input/06.caption +1 -0
- input/06.jpg +0 -0
- input/07.caption +1 -0
- input/07.jpg +0 -0
- input/08.caption +1 -0
- input/08.jpg +0 -0
- input/09.caption +1 -0
- input/09.jpg +0 -0
- logs/gen-turtle20231205105127/network_train/events.out.tfevents.1701773564.2392294ab23c.7086.0 +3 -0
- logs/gen-turtle20231205114254/network_train/events.out.tfevents.1701776626.2392294ab23c.19792.0 +3 -0
- output/gen-turtle-000005.safetensors +3 -0
- output/gen-turtle-000010.safetensors +3 -0
- output/gen-turtle-000015.safetensors +3 -0
- output/gen-turtle-000020.safetensors +3 -0
- output/sample/gen-turtle_20231205105734_e000001_01.png +0 -0
- output/sample/gen-turtle_20231205110225_e000002_01.png +0 -0
- output/sample/gen-turtle_20231205110716_e000003_01.png +0 -0
- output/sample/gen-turtle_20231205111207_e000004_01.png +0 -0
- output/sample/gen-turtle_20231205111657_e000005_01.png +0 -0
- output/sample/gen-turtle_20231205112149_e000006_01.png +0 -0
- output/sample/gen-turtle_20231205112640_e000007_01.png +0 -0
- output/sample/gen-turtle_20231205113131_e000008_01.png +0 -0
- output/sample/gen-turtle_20231205113622_e000009_01.png +0 -0
- output/sample/gen-turtle_20231205114113_e000010_01.png +0 -0
- output/sample/gen-turtle_20231205114839_e000001_01.png +0 -0
- output/sample/gen-turtle_20231205115331_e000002_01.png +0 -0
- output/sample/gen-turtle_20231205115822_e000003_01.png +0 -0
- output/sample/gen-turtle_20231205120313_e000004_01.png +0 -0
- output/sample/gen-turtle_20231205120805_e000005_01.png +0 -0
- output/sample/gen-turtle_20231205121256_e000006_01.png +0 -0
- output/sample/gen-turtle_20231205121747_e000007_01.png +0 -0
- output/sample/gen-turtle_20231205122239_e000008_01.png +0 -0
- output/sample/gen-turtle_20231205122730_e000009_01.png +0 -0
- output/sample/gen-turtle_20231205123221_e000010_01.png +0 -0
config/config_file.toml
ADDED
|
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[model_arguments]
|
| 2 |
+
v2 = false
|
| 3 |
+
v_parameterization = false
|
| 4 |
+
pretrained_model_name_or_path = "/content/drive/MyDrive/models/models/Stable-diffusion/AIGEN_1_4.ckpt"
|
| 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 |
+
network_weights = "/content/drive/MyDrive/LoRA/gen-turtle/output/gen-turtle-000010.safetensors"
|
| 16 |
+
|
| 17 |
+
[optimizer_arguments]
|
| 18 |
+
optimizer_type = "AdamW8bit"
|
| 19 |
+
learning_rate = 0.0003
|
| 20 |
+
max_grad_norm = 1.0
|
| 21 |
+
lr_scheduler = "constant"
|
| 22 |
+
lr_warmup_steps = 0
|
| 23 |
+
|
| 24 |
+
[dataset_arguments]
|
| 25 |
+
cache_latents = true
|
| 26 |
+
debug_dataset = false
|
| 27 |
+
vae_batch_size = 2
|
| 28 |
+
|
| 29 |
+
[training_arguments]
|
| 30 |
+
output_dir = "/content/drive/MyDrive/LoRA/gen-turtle/output"
|
| 31 |
+
output_name = "gen-turtle"
|
| 32 |
+
save_precision = "fp16"
|
| 33 |
+
save_every_n_epochs = 1
|
| 34 |
+
train_batch_size = 2
|
| 35 |
+
max_token_length = 75
|
| 36 |
+
mem_eff_attn = false
|
| 37 |
+
xformers = true
|
| 38 |
+
max_train_epochs = 10
|
| 39 |
+
max_data_loader_n_workers = 1
|
| 40 |
+
persistent_data_loader_workers = true
|
| 41 |
+
gradient_checkpointing = false
|
| 42 |
+
gradient_accumulation_steps = 1
|
| 43 |
+
mixed_precision = "fp16"
|
| 44 |
+
clip_skip = 2
|
| 45 |
+
logging_dir = "/content/drive/MyDrive/LoRA/gen-turtle/logs"
|
| 46 |
+
log_prefix = "gen-turtle"
|
| 47 |
+
lowram = true
|
| 48 |
+
|
| 49 |
+
[sample_prompt_arguments]
|
| 50 |
+
sample_every_n_epochs = 1
|
| 51 |
+
sample_sampler = "k_dpm_2"
|
| 52 |
+
|
| 53 |
+
[dreambooth_arguments]
|
| 54 |
+
prior_loss_weight = 1.0
|
| 55 |
+
|
| 56 |
+
[saving_arguments]
|
| 57 |
+
save_model_as = "safetensors"
|
config/dataset_config.toml
ADDED
|
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[[datasets]]
|
| 2 |
+
resolution = 512
|
| 3 |
+
min_bucket_reso = 256
|
| 4 |
+
max_bucket_reso = 1024
|
| 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/drive/MyDrive/LoRA/gen-turtle/input"
|
| 12 |
+
class_tokens = "turtle"
|
| 13 |
+
num_repeats = 100
|
| 14 |
+
|
| 15 |
+
|
| 16 |
+
[general]
|
| 17 |
+
enable_bucket = true
|
| 18 |
+
caption_extension = ".caption"
|
| 19 |
+
shuffle_caption = true
|
| 20 |
+
keep_tokens = 0
|
| 21 |
+
bucket_reso_steps = 64
|
| 22 |
+
bucket_no_upscale = false
|
config/sample_prompt.txt
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
(mdjrny-v4 style), masterpiece, best quality, turtle, sealife,ocean,underwater,beach --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 512 --l 7 --s 28
|
input/00.caption
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
a turtle swimming in the ocean with rocks and water
|
input/00.jpg
ADDED
|
input/01.caption
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
a turtle is sitting in the grass by the water
|
input/01.jpg
ADDED
|
input/02.caption
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
a turtle sitting on top of a wooden floor
|
input/02.jpg
ADDED
|
input/03.caption
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
a turtle swimming in the ocean with sun shining through the water
|
input/03.jpg
ADDED
|
input/04.caption
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
a small turtle crawling up a hill of sand
|
input/04.jpg
ADDED
|
input/05.caption
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
a turtle is walking through the grass
|
input/05.jpg
ADDED
|
input/06.caption
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
a baby turtle crawling on the sand on the beach
|
input/06.jpg
ADDED
|
input/07.caption
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
a close up of a tortoise with its mouth open
|
input/07.jpg
ADDED
|
input/08.caption
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
a turtle swimming in the ocean with a blue background
|
input/08.jpg
ADDED
|
input/09.caption
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
a turtle swimming in the ocean with a fish nearby
|
input/09.jpg
ADDED
|
logs/gen-turtle20231205105127/network_train/events.out.tfevents.1701773564.2392294ab23c.7086.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4ec603ee619422f20b0753639b304b5821e5abee4822b2bcf0fc35632223d723
|
| 3 |
+
size 1005060
|
logs/gen-turtle20231205114254/network_train/events.out.tfevents.1701776626.2392294ab23c.19792.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:93601ad6fa5db1740ac2d21cff5f6bf0a676a63bd2da30944f073ab094e1d274
|
| 3 |
+
size 1005060
|
output/gen-turtle-000005.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:28b6ba59cb1b711fc29079843def53a7bddf0c882dcce138a449726de6beef65
|
| 3 |
+
size 37864487
|
output/gen-turtle-000010.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:29715d6d3af75e2e7393c1d6f50717effec516a079c37c9dd5025258d649e950
|
| 3 |
+
size 37864488
|
output/gen-turtle-000015.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5d3b57e6cc6d4f7828f461b36493233a480e2e049cac6759160554e66bd10352
|
| 3 |
+
size 37864488
|
output/gen-turtle-000020.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1b6166d392407a655c783e27c3801af54b70961e8da28ca71e8c9d81561d13f6
|
| 3 |
+
size 37864488
|
output/sample/gen-turtle_20231205105734_e000001_01.png
ADDED
|
output/sample/gen-turtle_20231205110225_e000002_01.png
ADDED
|
output/sample/gen-turtle_20231205110716_e000003_01.png
ADDED
|
output/sample/gen-turtle_20231205111207_e000004_01.png
ADDED
|
output/sample/gen-turtle_20231205111657_e000005_01.png
ADDED
|
output/sample/gen-turtle_20231205112149_e000006_01.png
ADDED
|
output/sample/gen-turtle_20231205112640_e000007_01.png
ADDED
|
output/sample/gen-turtle_20231205113131_e000008_01.png
ADDED
|
output/sample/gen-turtle_20231205113622_e000009_01.png
ADDED
|
output/sample/gen-turtle_20231205114113_e000010_01.png
ADDED
|
output/sample/gen-turtle_20231205114839_e000001_01.png
ADDED
|
output/sample/gen-turtle_20231205115331_e000002_01.png
ADDED
|
output/sample/gen-turtle_20231205115822_e000003_01.png
ADDED
|
output/sample/gen-turtle_20231205120313_e000004_01.png
ADDED
|
output/sample/gen-turtle_20231205120805_e000005_01.png
ADDED
|
output/sample/gen-turtle_20231205121256_e000006_01.png
ADDED
|
output/sample/gen-turtle_20231205121747_e000007_01.png
ADDED
|
output/sample/gen-turtle_20231205122239_e000008_01.png
ADDED
|
output/sample/gen-turtle_20231205122730_e000009_01.png
ADDED
|
output/sample/gen-turtle_20231205123221_e000010_01.png
ADDED
|