ygohel18 commited on
Commit
0c92695
·
1 Parent(s): 2d1491e
Files changed (49) hide show
  1. config/config_file.toml +57 -0
  2. config/dataset_config.toml +22 -0
  3. config/sample_prompt.txt +2 -0
  4. input/00.caption +1 -0
  5. input/00.jpg +0 -0
  6. input/01.caption +1 -0
  7. input/01.jpg +0 -0
  8. input/02.caption +1 -0
  9. input/02.jpg +0 -0
  10. input/03.caption +1 -0
  11. input/03.jpg +0 -0
  12. input/04.caption +1 -0
  13. input/04.jpg +0 -0
  14. input/05.caption +1 -0
  15. input/05.jpg +0 -0
  16. input/06.caption +1 -0
  17. input/06.jpg +0 -0
  18. input/07.caption +1 -0
  19. input/07.jpg +0 -0
  20. input/08.caption +1 -0
  21. input/08.jpg +0 -0
  22. input/09.caption +1 -0
  23. input/09.jpg +0 -0
  24. logs/gen-turtle20231205105127/network_train/events.out.tfevents.1701773564.2392294ab23c.7086.0 +3 -0
  25. logs/gen-turtle20231205114254/network_train/events.out.tfevents.1701776626.2392294ab23c.19792.0 +3 -0
  26. output/gen-turtle-000005.safetensors +3 -0
  27. output/gen-turtle-000010.safetensors +3 -0
  28. output/gen-turtle-000015.safetensors +3 -0
  29. output/gen-turtle-000020.safetensors +3 -0
  30. output/sample/gen-turtle_20231205105734_e000001_01.png +0 -0
  31. output/sample/gen-turtle_20231205110225_e000002_01.png +0 -0
  32. output/sample/gen-turtle_20231205110716_e000003_01.png +0 -0
  33. output/sample/gen-turtle_20231205111207_e000004_01.png +0 -0
  34. output/sample/gen-turtle_20231205111657_e000005_01.png +0 -0
  35. output/sample/gen-turtle_20231205112149_e000006_01.png +0 -0
  36. output/sample/gen-turtle_20231205112640_e000007_01.png +0 -0
  37. output/sample/gen-turtle_20231205113131_e000008_01.png +0 -0
  38. output/sample/gen-turtle_20231205113622_e000009_01.png +0 -0
  39. output/sample/gen-turtle_20231205114113_e000010_01.png +0 -0
  40. output/sample/gen-turtle_20231205114839_e000001_01.png +0 -0
  41. output/sample/gen-turtle_20231205115331_e000002_01.png +0 -0
  42. output/sample/gen-turtle_20231205115822_e000003_01.png +0 -0
  43. output/sample/gen-turtle_20231205120313_e000004_01.png +0 -0
  44. output/sample/gen-turtle_20231205120805_e000005_01.png +0 -0
  45. output/sample/gen-turtle_20231205121256_e000006_01.png +0 -0
  46. output/sample/gen-turtle_20231205121747_e000007_01.png +0 -0
  47. output/sample/gen-turtle_20231205122239_e000008_01.png +0 -0
  48. output/sample/gen-turtle_20231205122730_e000009_01.png +0 -0
  49. 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