e10
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- config/config_file.toml +57 -0
- config/dataset_config.toml +22 -0
- config/sample_prompt.txt +1 -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
- input/10.caption +1 -0
- input/10.jpg +0 -0
- logs/gen-aurora20231205100109/network_train/events.out.tfevents.1701770523.d372b2d49408.10888.0 +3 -0
- logs/gen-aurora20231205100724/network_train/events.out.tfevents.1701770909.d372b2d49408.12477.0 +3 -0
- logs/gen-aurora20231205101025/network_train/events.out.tfevents.1701771088.d372b2d49408.13290.0 +3 -0
- logs/gen-aurora20231205101818/network_train/events.out.tfevents.1701771550.d372b2d49408.15335.0 +3 -0
- logs/gen-aurora20231205102337/network_train/events.out.tfevents.1701771862.d372b2d49408.16726.0 +3 -0
- logs/gen-aurora20231205111906/network_train/events.out.tfevents.1701775200.d372b2d49408.30561.0 +3 -0
- logs/last20231205095114/network_train/events.out.tfevents.1701769969.d372b2d49408.8288.0 +3 -0
- output/gen-aurora-000005.safetensors +3 -0
- output/gen-aurora-000010.safetensors +3 -0
- output/gen-aurora-000015.safetensors +3 -0
- output/gen-aurora-000020.safetensors +3 -0
- output/sample/gen-aurora_20231205102914_e000001_00.png +0 -0
- output/sample/gen-aurora_20231205103406_e000002_00.png +0 -0
- output/sample/gen-aurora_20231205103859_e000003_00.png +0 -0
- output/sample/gen-aurora_20231205104351_e000004_00.png +0 -0
- output/sample/gen-aurora_20231205104844_e000005_00.png +0 -0
- output/sample/gen-aurora_20231205105337_e000006_00.png +0 -0
- output/sample/gen-aurora_20231205105830_e000007_00.png +0 -0
- output/sample/gen-aurora_20231205110323_e000008_00.png +0 -0
- output/sample/gen-aurora_20231205110815_e000009_00.png +0 -0
- output/sample/gen-aurora_20231205111307_e000010_00.png +0 -0
- output/sample/gen-aurora_20231205112450_e000001_00.png +0 -0
- output/sample/gen-aurora_20231205112942_e000002_00.png +0 -0
- output/sample/gen-aurora_20231205113433_e000003_00.png +0 -0
- output/sample/gen-aurora_20231205113925_e000004_00.png +0 -0
- output/sample/gen-aurora_20231205114416_e000005_00.png +0 -0
- output/sample/gen-aurora_20231205114908_e000006_00.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-aurora/output/gen-aurora-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-aurora/output"
|
| 31 |
+
output_name = "gen-aurora"
|
| 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-aurora/logs"
|
| 46 |
+
log_prefix = "gen-aurora"
|
| 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-aurora/input"
|
| 12 |
+
class_tokens = "aurora"
|
| 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 @@
|
|
|
|
|
|
|
| 1 |
+
(mdjrny-v4 style), masterpiece, best quality, nature, mountains, snow, aurora, northan lights, alaska, river, nightsky --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/01.caption
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
a green aurora bore is in the sky
|
input/01.jpg
ADDED
|
input/02.caption
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
a green aurora bore over a frozen mountain
|
input/02.jpg
ADDED
|
input/03.caption
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
a green aurora bore over a mountain range
|
input/03.jpg
ADDED
|
input/04.caption
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
a road with a mountain in the background and a green and purple aurora bore
|
input/04.jpg
ADDED
|
input/05.caption
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
a green and purple aurora bore over a mountain range
|
input/05.jpg
ADDED
|
input/06.caption
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
a green and purple aurora bore over a body of water
|
input/06.jpg
ADDED
|
input/07.caption
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
a mountain with a green aurora bore in the sky
|
input/07.jpg
ADDED
|
input/08.caption
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
a green and red aurora bore over a body of water
|
input/08.jpg
ADDED
|
input/09.caption
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
a green and purple aurora bore over a body of water
|
input/09.jpg
ADDED
|
input/10.caption
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
a person standing on a snow covered slope under a green aurora
|
input/10.jpg
ADDED
|
logs/gen-aurora20231205100109/network_train/events.out.tfevents.1701770523.d372b2d49408.10888.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7b5cfa17102adcfc51a0128c19721004aad1cfa0339fb248e800c5f851309648
|
| 3 |
+
size 30210
|
logs/gen-aurora20231205100724/network_train/events.out.tfevents.1701770909.d372b2d49408.12477.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f3ce89626ae281768641a70b8c6c8f83ae1ec82a38a640ac9baf1609289cbd5e
|
| 3 |
+
size 13777
|
logs/gen-aurora20231205101025/network_train/events.out.tfevents.1701771088.d372b2d49408.13290.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3a45478ef7c875157173ad86229ddb1d283a8f963846864137a06220c97498ff
|
| 3 |
+
size 7031
|
logs/gen-aurora20231205101818/network_train/events.out.tfevents.1701771550.d372b2d49408.15335.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:abf058b05de9bc661ce21145471a83f7bc3f4e0ad7e259ed048514139aaea423
|
| 3 |
+
size 38220
|
logs/gen-aurora20231205102337/network_train/events.out.tfevents.1701771862.d372b2d49408.16726.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a9ba6764922f13b1f4020bd7117c9451dd9b3c19b5f83688be48c8a4bb33a43f
|
| 3 |
+
size 1005060
|
logs/gen-aurora20231205111906/network_train/events.out.tfevents.1701775200.d372b2d49408.30561.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8316abb120cfc6799531342f06606eeb5a18adff76a6884ed9ba08abfac2c27f
|
| 3 |
+
size 1005060
|
logs/last20231205095114/network_train/events.out.tfevents.1701769969.d372b2d49408.8288.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6f4836a7da589518c07b8dab990cd3a8d2cd14d7a067f6965afd40e50166772a
|
| 3 |
+
size 50070
|
output/gen-aurora-000005.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c3e8c716362af929ff3018200ecd92d6d6b9fcc25fdc55794bcef58f9a7bb091
|
| 3 |
+
size 37864424
|
output/gen-aurora-000010.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1f8cc116c2fff9173103b2bd33cee74397b6c8d178dfbf95bce4f273dac44ee8
|
| 3 |
+
size 37864427
|
output/gen-aurora-000015.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5d813712ff3f44eca33d0fe5484d1893ad4ef8e065ec2f4685bb3b35b6d02159
|
| 3 |
+
size 37864426
|
output/gen-aurora-000020.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e4210dc621ee5bcc9ca522330bfae9f9b89d4ae61c66aeaf408c775a79338be8
|
| 3 |
+
size 37864427
|
output/sample/gen-aurora_20231205102914_e000001_00.png
ADDED
|
output/sample/gen-aurora_20231205103406_e000002_00.png
ADDED
|
output/sample/gen-aurora_20231205103859_e000003_00.png
ADDED
|
output/sample/gen-aurora_20231205104351_e000004_00.png
ADDED
|
output/sample/gen-aurora_20231205104844_e000005_00.png
ADDED
|
output/sample/gen-aurora_20231205105337_e000006_00.png
ADDED
|
output/sample/gen-aurora_20231205105830_e000007_00.png
ADDED
|
output/sample/gen-aurora_20231205110323_e000008_00.png
ADDED
|
output/sample/gen-aurora_20231205110815_e000009_00.png
ADDED
|
output/sample/gen-aurora_20231205111307_e000010_00.png
ADDED
|
output/sample/gen-aurora_20231205112450_e000001_00.png
ADDED
|
output/sample/gen-aurora_20231205112942_e000002_00.png
ADDED
|
output/sample/gen-aurora_20231205113433_e000003_00.png
ADDED
|
output/sample/gen-aurora_20231205113925_e000004_00.png
ADDED
|
output/sample/gen-aurora_20231205114416_e000005_00.png
ADDED
|
output/sample/gen-aurora_20231205114908_e000006_00.png
ADDED
|