Sri2901 commited on
Commit
b41f6ea
·
verified ·
1 Parent(s): b9362aa

Upload model via CLI

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. .gitattributes +3 -0
  2. .job_config.json +114 -0
  3. 5_6_yrs_kids_shift.safetensors +3 -0
  4. 5_6_yrs_kids_shift_000000500.safetensors +3 -0
  5. 5_6_yrs_kids_shift_000001000.safetensors +3 -0
  6. 5_6_yrs_kids_shift_000001500.safetensors +3 -0
  7. 5_6_yrs_kids_shift_000002000.safetensors +3 -0
  8. 5_6_yrs_kids_shift_000002500.safetensors +3 -0
  9. 5_6_yrs_kids_shift_000003000.safetensors +3 -0
  10. 5_6_yrs_kids_shift_000003500.safetensors +3 -0
  11. README.md +102 -0
  12. config.yaml +138 -0
  13. log.txt +0 -0
  14. logs/0_log.txt +156 -0
  15. optimizer.pt +3 -0
  16. samples/1752325399103__000000000_0.jpg +0 -0
  17. samples/1752325416474__000000000_1.jpg +0 -0
  18. samples/1752325433898__000000000_2.jpg +0 -0
  19. samples/1752325451354__000000000_3.jpg +0 -0
  20. samples/1752325468722__000000000_4.jpg +0 -0
  21. samples/1752326818586__000000500_0.jpg +0 -0
  22. samples/1752326836014__000000500_1.jpg +0 -0
  23. samples/1752326853372__000000500_2.jpg +0 -0
  24. samples/1752326870754__000000500_3.jpg +0 -0
  25. samples/1752326888115__000000500_4.jpg +0 -0
  26. samples/1752328233409__000001000_0.jpg +0 -0
  27. samples/1752328250849__000001000_1.jpg +0 -0
  28. samples/1752328268220__000001000_2.jpg +0 -0
  29. samples/1752328285615__000001000_3.jpg +0 -0
  30. samples/1752328302974__000001000_4.jpg +0 -0
  31. samples/1752329664235__000001500_0.jpg +0 -0
  32. samples/1752329681651__000001500_1.jpg +3 -0
  33. samples/1752329699066__000001500_2.jpg +0 -0
  34. samples/1752329716471__000001500_3.jpg +0 -0
  35. samples/1752329733884__000001500_4.jpg +0 -0
  36. samples/1752331071225__000002000_0.jpg +0 -0
  37. samples/1752331088666__000002000_1.jpg +0 -0
  38. samples/1752331105956__000002000_2.jpg +0 -0
  39. samples/1752331123246__000002000_3.jpg +0 -0
  40. samples/1752331140536__000002000_4.jpg +3 -0
  41. samples/1752332519030__000002500_0.jpg +0 -0
  42. samples/1752332536457__000002500_1.jpg +0 -0
  43. samples/1752332553875__000002500_2.jpg +0 -0
  44. samples/1752332571317__000002500_3.jpg +0 -0
  45. samples/1752332588737__000002500_4.jpg +0 -0
  46. samples/1752333965576__000003000_0.jpg +0 -0
  47. samples/1752333983078__000003000_1.jpg +0 -0
  48. samples/1752334000581__000003000_2.jpg +0 -0
  49. samples/1752334017947__000003000_3.jpg +0 -0
  50. samples/1752334035459__000003000_4.jpg +0 -0
.gitattributes CHANGED
@@ -33,3 +33,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ samples/1752329681651__000001500_1.jpg filter=lfs diff=lfs merge=lfs -text
37
+ samples/1752331140536__000002000_4.jpg filter=lfs diff=lfs merge=lfs -text
38
+ samples/1752335444967__000003500_2.jpg filter=lfs diff=lfs merge=lfs -text
.job_config.json ADDED
@@ -0,0 +1,114 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "job": "extension",
3
+ "config": {
4
+ "name": "5_6_yrs_kids_shift",
5
+ "process": [
6
+ {
7
+ "type": "ui_trainer",
8
+ "training_folder": "/workspace/ai-toolkit/output",
9
+ "sqlite_db_path": "/workspace/ai-toolkit/aitk_db.db",
10
+ "device": "cuda",
11
+ "trigger_word": "k!ds",
12
+ "performance_log_every": 10,
13
+ "network": {
14
+ "type": "lora",
15
+ "linear": 64,
16
+ "linear_alpha": 64,
17
+ "conv": 16,
18
+ "conv_alpha": 16,
19
+ "lokr_full_rank": true,
20
+ "lokr_factor": -1,
21
+ "network_kwargs": {
22
+ "ignore_if_contains": []
23
+ }
24
+ },
25
+ "save": {
26
+ "dtype": "bf16",
27
+ "save_every": 500,
28
+ "max_step_saves_to_keep": 8,
29
+ "save_format": "diffusers",
30
+ "push_to_hub": false
31
+ },
32
+ "datasets": [
33
+ {
34
+ "folder_path": "/workspace/ai-toolkit/datasets/5_6_yrs_kids",
35
+ "control_path": null,
36
+ "mask_path": null,
37
+ "mask_min_value": 0.1,
38
+ "default_caption": "",
39
+ "caption_ext": "txt",
40
+ "caption_dropout_rate": 0.05,
41
+ "cache_latents_to_disk": false,
42
+ "is_reg": false,
43
+ "network_weight": 1,
44
+ "resolution": [
45
+ 512,
46
+ 768,
47
+ 1024,
48
+ 1280
49
+ ],
50
+ "controls": []
51
+ }
52
+ ],
53
+ "train": {
54
+ "batch_size": 1,
55
+ "bypass_guidance_embedding": false,
56
+ "steps": 4000,
57
+ "gradient_accumulation": 1,
58
+ "train_unet": true,
59
+ "train_text_encoder": false,
60
+ "gradient_checkpointing": true,
61
+ "noise_scheduler": "flowmatch",
62
+ "optimizer": "adamw8bit",
63
+ "timestep_type": "shift",
64
+ "content_or_style": "balanced",
65
+ "optimizer_params": {
66
+ "weight_decay": 0.0001
67
+ },
68
+ "unload_text_encoder": false,
69
+ "lr": 0.0002,
70
+ "ema_config": {
71
+ "use_ema": false,
72
+ "ema_decay": 0.99
73
+ },
74
+ "dtype": "bf16",
75
+ "diff_output_preservation": false,
76
+ "diff_output_preservation_multiplier": 1,
77
+ "diff_output_preservation_class": "person"
78
+ },
79
+ "model": {
80
+ "name_or_path": "black-forest-labs/FLUX.1-dev",
81
+ "quantize": false,
82
+ "quantize_te": true,
83
+ "arch": "flux",
84
+ "low_vram": false,
85
+ "model_kwargs": {}
86
+ },
87
+ "sample": {
88
+ "sampler": "flowmatch",
89
+ "sample_every": 500,
90
+ "width": 1024,
91
+ "height": 1024,
92
+ "prompts": [
93
+ "A 5-year old girl with blonde hair, standing in the middle of the frame, wearing a green knitted top with a geometric pattern and a white floral skirt with ruffled layers. she has a playful expression and is looking directly at the camera with a slight smile on her face. the background is a plain white wall with a shadow cast on it, creating a simple yet elegant atmosphere. the girl appears to be around 6 years old, with her hair styled in a neat, straight look and her eyes looking directly into the camera. her outfit consists of a sleeveless dress with a green and white geometric pattern, paired with a pair of white floral pants. the overall aesthetic of the image is playful and cheerful, with a focus on the girl's cheerful expression and the texture of her outfit.",
94
+ "a 5-year-old girl, who appears to be smiling widely, is standing with her arms outstretched. she has dark skin, styled in a single hair bun, and is wearing a sleeveless, knitted dress with a colorful floral pattern. the dress has a square neckline and thin straps, and she is also wearing a pair of earrings. her expression is one of joy and laughter, with her mouth open and teeth showing. her eyes are looking directly at the viewer, and her ears are not visible in the image.",
95
+ "Three young 4 year old children standing in a row against a bright blue sky, smiling and looking directly at the camera. on the left, a young girl with blonde hair stands in the middle, wearing a white dress with short sleeves and a floral pattern, while on the right, a younger boy stands behind her, also wearing white pants with a similar pattern. the younger boy has short blonde hair and is standing with his hands clasped together in front of him, while the older boy is standing behind him with his arms around his waist. all three children have a cheerful expression on their faces, with their faces facing the same direction and their eyes looking straight at the viewer. the blue sky provides a clear backdrop for the children, and the overall mood of the image is bright and cheerful, conveying a sense of unity and togetherness.",
96
+ "A photo-realistic shoot from the front about a young boy sitting on a white cube wearing a green knitted shirt and white shorts. the image also shows a neutral color palette and minimalistic background. on the middle of the image, a 6-year-old boy with curly black hair and a neutral expression sits on a square white cube. he is wearing a short-sleeved, light green knit shirt with a button-down collar and short sleeves, paired with white denim shorts. his hands are clasped together in front of him, and he is looking directly at the camera with a slight smile. he has brown eyes and is wearing tan suede mules on his feet. the boy is sitting on the cube in a relaxed pose, with his legs slightly apart and his hands resting on his knees. the lighting is soft and natural, casting gentle shadows on his face and body.",
97
+ "A portrait shoot from a portrait camera angle about a 5 year old young boy with curly black hair wearing a light green cardigan, looking directly at the viewer with a playful expression. the image also shows a plain white background. on the middle of the image, a 5-year-old boy, with a neutral expression, is facing the camera directly. he is wearing a green knitted cardigan with a button-down collar, which is slightly blurred to emphasize the boy's features. the boy has dark brown curly hair and brown eyes, which are looking straight into the camera with a slight smile on his lips. his body is slim, and he is standing in front of the camera, with his upper body visible."
98
+ ],
99
+ "neg": "",
100
+ "seed": 42,
101
+ "walk_seed": true,
102
+ "guidance_scale": 3,
103
+ "sample_steps": 30,
104
+ "num_frames": 1,
105
+ "fps": 1
106
+ }
107
+ }
108
+ ]
109
+ },
110
+ "meta": {
111
+ "name": "[name]",
112
+ "version": "1.0"
113
+ }
114
+ }
5_6_yrs_kids_shift.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:149edc3a5712d46a7e0d9706126c4ce6920fcabec17a4da437a97e0e96998a7f
3
+ size 687477552
5_6_yrs_kids_shift_000000500.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bd8177f8c1cafe7dfa690f0c3eb2b8f05511ac2d5e063e8ee98940737a8aeb6a
3
+ size 687477552
5_6_yrs_kids_shift_000001000.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6e6ad0f24243b74ed00a85f7776d3eb16ad732a0c01ea73a15c63f955004f2bb
3
+ size 687477552
5_6_yrs_kids_shift_000001500.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e3252e70a7bc592dbf2a0a1b63b3d60e716d54fcfb57b5fbe4c36e5621a9d16f
3
+ size 687477552
5_6_yrs_kids_shift_000002000.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a89c5088c8f6f3772eed6ecaeb4cfbdd2ab200f52512ee9625e37d3c580337c8
3
+ size 687477552
5_6_yrs_kids_shift_000002500.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:812582f7bf7779dfd8e09ef60d4a733961e4557d472a0a12eb478bfbe06dc191
3
+ size 687477552
5_6_yrs_kids_shift_000003000.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3b11730d6dbcd5120e50908b920734b7d695a92079d76337be33fc6ce0c53cbb
3
+ size 687477552
5_6_yrs_kids_shift_000003500.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:41e20d6443933030487e0ba34c548c5a6801e269040d8b71723e6b2cef7ca100
3
+ size 687477552
README.md ADDED
@@ -0,0 +1,102 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - text-to-image
4
+ - flux
5
+ - lora
6
+ - diffusers
7
+ - template:sd-lora
8
+ - ai-toolkit
9
+ base_model: black-forest-labs/FLUX.1-dev
10
+ instance_prompt: k!ds
11
+ license: other
12
+ license_name: flux-1-dev-non-commercial-license
13
+ license_link: https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md
14
+ widget:
15
+ - text: A 5-year old girl with blonde hair, standing in the middle of the frame, wearing
16
+ a green knitted top with a geometric pattern and a white floral skirt with ruffled
17
+ layers. she has a playful expression and is looking directly at the camera with
18
+ a slight smile on her face. the background is a plain white wall with a shadow
19
+ cast on it, creating a simple yet elegant atmosphere. the girl appears to be around
20
+ 6 years old, with her hair styled in a neat, straight look and her eyes looking
21
+ directly into the camera. her outfit consists of a sleeveless dress with a green
22
+ and white geometric pattern, paired with a pair of white floral pants. the overall
23
+ aesthetic of the image is playful and cheerful, with a focus on the girl's cheerful
24
+ expression and the texture of her outfit.
25
+ output:
26
+ url: samples/1752336921059__000004000_4.jpg
27
+ - text: a 5-year-old girl, who appears to be smiling widely, is standing with her
28
+ arms outstretched. she has dark skin, styled in a single hair bun, and is wearing
29
+ a sleeveless, knitted dress with a colorful floral pattern. the dress has a square
30
+ neckline and thin straps, and she is also wearing a pair of earrings. her expression
31
+ is one of joy and laughter, with her mouth open and teeth showing. her eyes are
32
+ looking directly at the viewer, and her ears are not visible in the image.
33
+ output:
34
+ url: samples/1752336903481__000004000_3.jpg
35
+ - text: Three young 4 year old children standing in a row against a bright blue sky,
36
+ smiling and looking directly at the camera. on the left, a young girl with blonde
37
+ hair stands in the middle, wearing a white dress with short sleeves and a floral
38
+ pattern, while on the right, a younger boy stands behind her, also wearing white
39
+ pants with a similar pattern. the younger boy has short blonde hair and is standing
40
+ with his hands clasped together in front of him, while the older boy is standing
41
+ behind him with his arms around his waist. all three children have a cheerful
42
+ expression on their faces, with their faces facing the same direction and their
43
+ eyes looking straight at the viewer. the blue sky provides a clear backdrop for
44
+ the children, and the overall mood of the image is bright and cheerful, conveying
45
+ a sense of unity and togetherness.
46
+ output:
47
+ url: samples/1752336885908__000004000_2.jpg
48
+ - text: A photo-realistic shoot from the front about a young boy sitting on a white
49
+ cube wearing a green knitted shirt and white shorts. the image also shows a neutral
50
+ color palette and minimalistic background. on the middle of the image, a 6-year-old
51
+ boy with curly black hair and a neutral expression sits on a square white cube.
52
+ he is wearing a short-sleeved, light green knit shirt with a button-down collar
53
+ and short sleeves, paired with white denim shorts. his hands are clasped together
54
+ in front of him, and he is looking directly at the camera with a slight smile.
55
+ he has brown eyes and is wearing tan suede mules on his feet. the boy is sitting
56
+ on the cube in a relaxed pose, with his legs slightly apart and his hands resting
57
+ on his knees. the lighting is soft and natural, casting gentle shadows on his
58
+ face and body.
59
+ output:
60
+ url: samples/1752336868349__000004000_1.jpg
61
+ - text: A portrait shoot from a portrait camera angle about a 5 year old young boy
62
+ with curly black hair wearing a light green cardigan, looking directly at the
63
+ viewer with a playful expression. the image also shows a plain white background.
64
+ on the middle of the image, a 5-year-old boy, with a neutral expression, is facing
65
+ the camera directly. he is wearing a green knitted cardigan with a button-down
66
+ collar, which is slightly blurred to emphasize the boy's features. the boy has
67
+ dark brown curly hair and brown eyes, which are looking straight into the camera
68
+ with a slight smile on his lips. his body is slim, and he is standing in front
69
+ of the camera, with his upper body visible.
70
+ output:
71
+ url: samples/1752336850762__000004000_0.jpg
72
+ ---
73
+
74
+ # 5_6_yrs_kids_shift
75
+
76
+ Model trained with AI Toolkit by Ostris
77
+
78
+ <Gallery />
79
+
80
+ ## Trigger words
81
+
82
+ You should use `k!ds` to trigger the image generation.
83
+
84
+ ## Download model and use it with ComfyUI, AUTOMATIC1111, SD.Next, Invoke AI, etc.
85
+
86
+ Weights for this model are available in Safetensors format.
87
+
88
+ [Download](/username/5_6_yrs_kids_shift/tree/main) them in the Files & versions tab.
89
+
90
+ ## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
91
+
92
+ ```py
93
+ from diffusers import AutoPipelineForText2Image
94
+ import torch
95
+
96
+ pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.bfloat16).to('cuda')
97
+ pipeline.load_lora_weights('username/5_6_yrs_kids_shift', weight_name='5_6_yrs_kids_shift.safetensors')
98
+ image = pipeline('k!ds A 5-year old girl with blonde hair, standing in the middle of the frame, wearing a green knitted top with a geometric pattern and a white floral skirt with ruffled layers. she has a playful expression and is looking directly at the camera with a slight smile on her face. the background is a plain white wall with a shadow cast on it, creating a simple yet elegant atmosphere. the girl appears to be around 6 years old, with her hair styled in a neat, straight look and her eyes looking directly into the camera. her outfit consists of a sleeveless dress with a green and white geometric pattern, paired with a pair of white floral pants. the overall aesthetic of the image is playful and cheerful, with a focus on the girl's cheerful expression and the texture of her outfit.').images[0]
99
+ image.save("my_image.png")
100
+ ```
101
+
102
+ For more details, including weighting, merging and fusing LoRAs, check the [documentation on loading LoRAs in diffusers](https://huggingface.co/docs/diffusers/main/en/using-diffusers/loading_adapters)
config.yaml ADDED
@@ -0,0 +1,138 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ job: extension
2
+ config:
3
+ name: 5_6_yrs_kids_shift
4
+ process:
5
+ - type: ui_trainer
6
+ training_folder: /workspace/ai-toolkit/output
7
+ sqlite_db_path: /workspace/ai-toolkit/aitk_db.db
8
+ device: cuda
9
+ trigger_word: k!ds
10
+ performance_log_every: 10
11
+ network:
12
+ type: lora
13
+ linear: 64
14
+ linear_alpha: 64
15
+ conv: 16
16
+ conv_alpha: 16
17
+ lokr_full_rank: true
18
+ lokr_factor: -1
19
+ network_kwargs:
20
+ ignore_if_contains: []
21
+ save:
22
+ dtype: bf16
23
+ save_every: 500
24
+ max_step_saves_to_keep: 8
25
+ save_format: diffusers
26
+ push_to_hub: false
27
+ datasets:
28
+ - folder_path: /workspace/ai-toolkit/datasets/5_6_yrs_kids
29
+ control_path: null
30
+ mask_path: null
31
+ mask_min_value: 0.1
32
+ default_caption: ''
33
+ caption_ext: txt
34
+ caption_dropout_rate: 0.05
35
+ cache_latents_to_disk: false
36
+ is_reg: false
37
+ network_weight: 1
38
+ resolution:
39
+ - 512
40
+ - 768
41
+ - 1024
42
+ - 1280
43
+ controls: []
44
+ train:
45
+ batch_size: 1
46
+ bypass_guidance_embedding: false
47
+ steps: 4000
48
+ gradient_accumulation: 1
49
+ train_unet: true
50
+ train_text_encoder: false
51
+ gradient_checkpointing: true
52
+ noise_scheduler: flowmatch
53
+ optimizer: adamw8bit
54
+ timestep_type: shift
55
+ content_or_style: balanced
56
+ optimizer_params:
57
+ weight_decay: 0.0001
58
+ unload_text_encoder: false
59
+ lr: 0.0002
60
+ ema_config:
61
+ use_ema: false
62
+ ema_decay: 0.99
63
+ dtype: bf16
64
+ diff_output_preservation: false
65
+ diff_output_preservation_multiplier: 1
66
+ diff_output_preservation_class: person
67
+ model:
68
+ name_or_path: black-forest-labs/FLUX.1-dev
69
+ quantize: false
70
+ quantize_te: true
71
+ arch: flux
72
+ low_vram: false
73
+ model_kwargs: {}
74
+ sample:
75
+ sampler: flowmatch
76
+ sample_every: 500
77
+ width: 1024
78
+ height: 1024
79
+ prompts:
80
+ - A 5-year old girl with blonde hair, standing in the middle of the frame, wearing
81
+ a green knitted top with a geometric pattern and a white floral skirt with
82
+ ruffled layers. she has a playful expression and is looking directly at the
83
+ camera with a slight smile on her face. the background is a plain white wall
84
+ with a shadow cast on it, creating a simple yet elegant atmosphere. the girl
85
+ appears to be around 6 years old, with her hair styled in a neat, straight
86
+ look and her eyes looking directly into the camera. her outfit consists of
87
+ a sleeveless dress with a green and white geometric pattern, paired with a
88
+ pair of white floral pants. the overall aesthetic of the image is playful
89
+ and cheerful, with a focus on the girl's cheerful expression and the texture
90
+ of her outfit.
91
+ - a 5-year-old girl, who appears to be smiling widely, is standing with her
92
+ arms outstretched. she has dark skin, styled in a single hair bun, and is
93
+ wearing a sleeveless, knitted dress with a colorful floral pattern. the dress
94
+ has a square neckline and thin straps, and she is also wearing a pair of earrings.
95
+ her expression is one of joy and laughter, with her mouth open and teeth showing.
96
+ her eyes are looking directly at the viewer, and her ears are not visible
97
+ in the image.
98
+ - Three young 4 year old children standing in a row against a bright blue sky,
99
+ smiling and looking directly at the camera. on the left, a young girl with
100
+ blonde hair stands in the middle, wearing a white dress with short sleeves
101
+ and a floral pattern, while on the right, a younger boy stands behind her,
102
+ also wearing white pants with a similar pattern. the younger boy has short
103
+ blonde hair and is standing with his hands clasped together in front of him,
104
+ while the older boy is standing behind him with his arms around his waist.
105
+ all three children have a cheerful expression on their faces, with their faces
106
+ facing the same direction and their eyes looking straight at the viewer. the
107
+ blue sky provides a clear backdrop for the children, and the overall mood
108
+ of the image is bright and cheerful, conveying a sense of unity and togetherness.
109
+ - A photo-realistic shoot from the front about a young boy sitting on a white
110
+ cube wearing a green knitted shirt and white shorts. the image also shows
111
+ a neutral color palette and minimalistic background. on the middle of the
112
+ image, a 6-year-old boy with curly black hair and a neutral expression sits
113
+ on a square white cube. he is wearing a short-sleeved, light green knit shirt
114
+ with a button-down collar and short sleeves, paired with white denim shorts.
115
+ his hands are clasped together in front of him, and he is looking directly
116
+ at the camera with a slight smile. he has brown eyes and is wearing tan suede
117
+ mules on his feet. the boy is sitting on the cube in a relaxed pose, with
118
+ his legs slightly apart and his hands resting on his knees. the lighting is
119
+ soft and natural, casting gentle shadows on his face and body.
120
+ - A portrait shoot from a portrait camera angle about a 5 year old young boy
121
+ with curly black hair wearing a light green cardigan, looking directly at
122
+ the viewer with a playful expression. the image also shows a plain white background.
123
+ on the middle of the image, a 5-year-old boy, with a neutral expression, is
124
+ facing the camera directly. he is wearing a green knitted cardigan with a
125
+ button-down collar, which is slightly blurred to emphasize the boy's features.
126
+ the boy has dark brown curly hair and brown eyes, which are looking straight
127
+ into the camera with a slight smile on his lips. his body is slim, and he
128
+ is standing in front of the camera, with his upper body visible.
129
+ neg: ''
130
+ seed: 42
131
+ walk_seed: true
132
+ guidance_scale: 3
133
+ sample_steps: 30
134
+ num_frames: 1
135
+ fps: 1
136
+ meta:
137
+ name: 5_6_yrs_kids_shift
138
+ version: '1.0'
log.txt ADDED
The diff for this file is too large to render. See raw diff
 
logs/0_log.txt ADDED
@@ -0,0 +1,156 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Running 1 job
2
+ /workspace/ai-toolkit/venv/lib/python3.10/site-packages/albumentations/__init__.py:13: UserWarning: A new version of Albumentations is available: 2.0.8 (you have 1.4.15). Upgrade using: pip install -U albumentations. To disable automatic update checks, set the environment variable NO_ALBUMENTATIONS_UPDATE to 1.
3
+ check_for_updates()
4
+ /workspace/ai-toolkit/venv/lib/python3.10/site-packages/albumentations/__init__.py:13: UserWarning: A new version of Albumentations is available: 2.0.8 (you have 1.4.15). Upgrade using: pip install -U albumentations. To disable automatic update checks, set the environment variable NO_ALBUMENTATIONS_UPDATE to 1.
5
+ check_for_updates()
6
+ {
7
+ "type": "ui_trainer",
8
+ "training_folder": "/workspace/ai-toolkit/output",
9
+ "sqlite_db_path": "/workspace/ai-toolkit/aitk_db.db",
10
+ "device": "cuda",
11
+ "trigger_word": "k!ds",
12
+ "performance_log_every": 10,
13
+ "network": {
14
+ "type": "lora",
15
+ "linear": 64,
16
+ "linear_alpha": 64,
17
+ "conv": 16,
18
+ "conv_alpha": 16,
19
+ "lokr_full_rank": true,
20
+ "lokr_factor": -1,
21
+ "network_kwargs": {
22
+ "ignore_if_contains": []
23
+ }
24
+ },
25
+ "save": {
26
+ "dtype": "bf16",
27
+ "save_every": 500,
28
+ "max_step_saves_to_keep": 8,
29
+ "save_format": "diffusers",
30
+ "push_to_hub": false
31
+ },
32
+ "datasets": [
33
+ {
34
+ "folder_path": "/workspace/ai-toolkit/datasets/5_6_yrs_kids",
35
+ "control_path": null,
36
+ "mask_path": null,
37
+ "mask_min_value": 0.1,
38
+ "default_caption": "",
39
+ "caption_ext": "txt",
40
+ "caption_dropout_rate": 0.05,
41
+ "cache_latents_to_disk": false,
42
+ "is_reg": false,
43
+ "network_weight": 1,
44
+ "resolution": [
45
+ 512,
46
+ 768,
47
+ 1024,
48
+ 1280
49
+ ],
50
+ "controls": []
51
+ }
52
+ ],
53
+ "train": {
54
+ "batch_size": 1,
55
+ "bypass_guidance_embedding": false,
56
+ "steps": 4000,
57
+ "gradient_accumulation": 1,
58
+ "train_unet": true,
59
+ "train_text_encoder": false,
60
+ "gradient_checkpointing": true,
61
+ "noise_scheduler": "flowmatch",
62
+ "optimizer": "adamw8bit",
63
+ "timestep_type": "shift",
64
+ "content_or_style": "balanced",
65
+ "optimizer_params": {
66
+ "weight_decay": 0.0001
67
+ },
68
+ "unload_text_encoder": false,
69
+ "lr": 0.0002,
70
+ "ema_config": {
71
+ "use_ema": false,
72
+ "ema_decay": 0.99
73
+ },
74
+ "dtype": "bf16",
75
+ "diff_output_preservation": false,
76
+ "diff_output_preservation_multiplier": 1,
77
+ "diff_output_preservation_class": "person"
78
+ },
79
+ "model": {
80
+ "name_or_path": "black-forest-labs/FLUX.1-dev",
81
+ "quantize": false,
82
+ "quantize_te": true,
83
+ "arch": "flux",
84
+ "low_vram": false,
85
+ "model_kwargs": {}
86
+ },
87
+ "sample": {
88
+ "sampler": "flowmatch",
89
+ "sample_every": 500,
90
+ "width": 1024,
91
+ "height": 1024,
92
+ "prompts": [
93
+ "A 5-year old girl with blonde hair, standing in the middle of the frame, wearing a green knitted top with a geometric pattern and a white floral skirt with ruffled layers. she has a playful expression and is looking directly at the camera with a slight smile on her face. the background is a plain white wall with a shadow cast on it, creating a simple yet elegant atmosphere. the girl appears to be around 6 years old, with her hair styled in a neat, straight look and her eyes looking directly into the camera. her outfit consists of a sleeveless dress with a green and white geometric pattern, paired with a pair of white floral pants. the overall aesthetic of the image is playful and cheerful, with a focus on the girl's cheerful expression and the texture of her outfit.",
94
+ "a 5-year-old girl, who appears to be smiling widely, is standing with her arms outstretched. she has dark skin, styled in a single hair bun, and is wearing a sleeveless, knitted dress with a colorful floral pattern. the dress has a square neckline and thin straps, and she is also wearing a pair of earrings. her expression is one of joy and laughter, with her mouth open and teeth showing. her eyes are looking directly at the viewer, and her ears are not visible in the image.",
95
+ "Three young 4 year old children standing in a row against a bright blue sky, smiling and looking directly at the camera. on the left, a young girl with blonde hair stands in the middle, wearing a white dress with short sleeves and a floral pattern, while on the right, a younger boy stands behind her, also wearing white pants with a similar pattern. the younger boy has short blonde hair and is standing with his hands clasped together in front of him, while the older boy is standing behind him with his arms around his waist. all three children have a cheerful expression on their faces, with their faces facing the same direction and their eyes looking straight at the viewer. the blue sky provides a clear backdrop for the children, and the overall mood of the image is bright and cheerful, conveying a sense of unity and togetherness.",
96
+ "A photo-realistic shoot from the front about a young boy sitting on a white cube wearing a green knitted shirt and white shorts. the image also shows a neutral color palette and minimalistic background. on the middle of the image, a 6-year-old boy with curly black hair and a neutral expression sits on a square white cube. he is wearing a short-sleeved, light green knit shirt with a button-down collar and short sleeves, paired with white denim shorts. his hands are clasped together in front of him, and he is looking directly at the camera with a slight smile. he has brown eyes and is wearing tan suede mules on his feet. the boy is sitting on the cube in a relaxed pose, with his legs slightly apart and his hands resting on his knees. the lighting is soft and natural, casting gentle shadows on his face and body.",
97
+ "A portrait shoot from a portrait camera angle about a 5 year old young boy with curly black hair wearing a light green cardigan, looking directly at the viewer with a playful expression. the image also shows a plain white background. on the middle of the image, a 5-year-old boy, with a neutral expression, is facing the camera directly. he is wearing a green knitted cardigan with a button-down collar, which is slightly blurred to emphasize the boy's features. the boy has dark brown curly hair and brown eyes, which are looking straight into the camera with a slight smile on his lips. his body is slim, and he is standing in front of the camera, with his upper body visible."
98
+ ],
99
+ "neg": "",
100
+ "seed": 42,
101
+ "walk_seed": true,
102
+ "guidance_scale": 3,
103
+ "sample_steps": 30,
104
+ "num_frames": 1,
105
+ "fps": 1
106
+ }
107
+ }
108
+ Using EMA
109
+ Using SQLite database at /workspace/ai-toolkit/aitk_db.db
110
+ Job ID: "8f2b6ec7-5367-4eba-aa99-749a9ef24a32"
111
+
112
+ #############################################
113
+ # Running job: 5_6_yrs_kids_shift
114
+ #############################################
115
+
116
+
117
+ Running 1 process
118
+ Error running job: Job stopped
119
+
120
+ ========================================
121
+ Result:
122
+ - 0 completed jobs
123
+ - 1 failure
124
+ ========================================
125
+ Traceback (most recent call last):
126
+ File "/workspace/ai-toolkit/run.py", line 119, in <module>
127
+ main()
128
+ File "/workspace/ai-toolkit/run.py", line 107, in main
129
+ raise e
130
+ File "/workspace/ai-toolkit/run.py", line 95, in main
131
+ job.run()
132
+ File "/workspace/ai-toolkit/jobs/ExtensionJob.py", line 22, in run
133
+ process.run()
134
+ File "/workspace/ai-toolkit/jobs/process/BaseSDTrainProcess.py", line 1460, in run
135
+ self.hook_before_model_load()
136
+ File "/workspace/ai-toolkit/extensions_built_in/sd_trainer/UITrainer.py", line 192, in hook_before_model_load
137
+ self.maybe_stop()
138
+ File "/workspace/ai-toolkit/extensions_built_in/sd_trainer/UITrainer.py", line 79, in maybe_stop
139
+ raise Exception("Job stopped")
140
+ Exception: Job stopped
141
+ Traceback (most recent call last):
142
+ File "/workspace/ai-toolkit/run.py", line 119, in <module>
143
+ main()
144
+ File "/workspace/ai-toolkit/run.py", line 107, in main
145
+ raise e
146
+ File "/workspace/ai-toolkit/run.py", line 95, in main
147
+ job.run()
148
+ File "/workspace/ai-toolkit/jobs/ExtensionJob.py", line 22, in run
149
+ process.run()
150
+ File "/workspace/ai-toolkit/jobs/process/BaseSDTrainProcess.py", line 1460, in run
151
+ self.hook_before_model_load()
152
+ File "/workspace/ai-toolkit/extensions_built_in/sd_trainer/UITrainer.py", line 192, in hook_before_model_load
153
+ self.maybe_stop()
154
+ File "/workspace/ai-toolkit/extensions_built_in/sd_trainer/UITrainer.py", line 79, in maybe_stop
155
+ raise Exception("Job stopped")
156
+ Exception: Job stopped
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7cfbe5677c5a209bcbcea30db57da5ba9a305b41115fba7bd5668680d0577a09
3
+ size 699243652
samples/1752325399103__000000000_0.jpg ADDED
samples/1752325416474__000000000_1.jpg ADDED
samples/1752325433898__000000000_2.jpg ADDED
samples/1752325451354__000000000_3.jpg ADDED
samples/1752325468722__000000000_4.jpg ADDED
samples/1752326818586__000000500_0.jpg ADDED
samples/1752326836014__000000500_1.jpg ADDED
samples/1752326853372__000000500_2.jpg ADDED
samples/1752326870754__000000500_3.jpg ADDED
samples/1752326888115__000000500_4.jpg ADDED
samples/1752328233409__000001000_0.jpg ADDED
samples/1752328250849__000001000_1.jpg ADDED
samples/1752328268220__000001000_2.jpg ADDED
samples/1752328285615__000001000_3.jpg ADDED
samples/1752328302974__000001000_4.jpg ADDED
samples/1752329664235__000001500_0.jpg ADDED
samples/1752329681651__000001500_1.jpg ADDED

Git LFS Details

  • SHA256: ac537d9ed132fcb3c806e1cd9b0a3c779bc69794a9371f2156340c09d866fe57
  • Pointer size: 131 Bytes
  • Size of remote file: 101 kB
samples/1752329699066__000001500_2.jpg ADDED
samples/1752329716471__000001500_3.jpg ADDED
samples/1752329733884__000001500_4.jpg ADDED
samples/1752331071225__000002000_0.jpg ADDED
samples/1752331088666__000002000_1.jpg ADDED
samples/1752331105956__000002000_2.jpg ADDED
samples/1752331123246__000002000_3.jpg ADDED
samples/1752331140536__000002000_4.jpg ADDED

Git LFS Details

  • SHA256: cfe2c2464bb5d0cf5643806a9871c3f77fc9c8c60b619465a65a56c84e03cc62
  • Pointer size: 131 Bytes
  • Size of remote file: 101 kB
samples/1752332519030__000002500_0.jpg ADDED
samples/1752332536457__000002500_1.jpg ADDED
samples/1752332553875__000002500_2.jpg ADDED
samples/1752332571317__000002500_3.jpg ADDED
samples/1752332588737__000002500_4.jpg ADDED
samples/1752333965576__000003000_0.jpg ADDED
samples/1752333983078__000003000_1.jpg ADDED
samples/1752334000581__000003000_2.jpg ADDED
samples/1752334017947__000003000_3.jpg ADDED
samples/1752334035459__000003000_4.jpg ADDED