svjack commited on
Commit
8f88900
·
verified ·
1 Parent(s): 49f11d9

Upload folder using huggingface_hub

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 +41 -0
  2. .job_config.json +116 -0
  3. collei.safetensors +3 -0
  4. collei_000000250.safetensors +3 -0
  5. collei_000000500.safetensors +3 -0
  6. collei_000000750.safetensors +3 -0
  7. collei_000001000.safetensors +3 -0
  8. collei_000001250.safetensors +3 -0
  9. collei_000001500.safetensors +3 -0
  10. collei_000001750.safetensors +3 -0
  11. collei_000002000.safetensors +3 -0
  12. collei_000002250.safetensors +3 -0
  13. collei_000002500.safetensors +3 -0
  14. collei_000002750.safetensors +3 -0
  15. collei_000003000.safetensors +3 -0
  16. collei_000003250.safetensors +3 -0
  17. collei_000003500.safetensors +3 -0
  18. collei_000003750.safetensors +3 -0
  19. collei_000004000.safetensors +3 -0
  20. collei_000004250.safetensors +3 -0
  21. collei_000004500.safetensors +3 -0
  22. collei_000004750.safetensors +3 -0
  23. collei_000005000.safetensors +3 -0
  24. collei_000005250.safetensors +3 -0
  25. collei_000005500.safetensors +3 -0
  26. collei_000005750.safetensors +3 -0
  27. collei_000006000.safetensors +3 -0
  28. collei_000006250.safetensors +3 -0
  29. collei_000006500.safetensors +3 -0
  30. collei_000006750.safetensors +3 -0
  31. collei_000007000.safetensors +3 -0
  32. collei_000007250.safetensors +3 -0
  33. collei_000007500.safetensors +3 -0
  34. collei_000007750.safetensors +3 -0
  35. collei_000008000.safetensors +3 -0
  36. collei_000008250.safetensors +3 -0
  37. collei_000008500.safetensors +3 -0
  38. collei_000008750.safetensors +3 -0
  39. collei_000009000.safetensors +3 -0
  40. collei_000009250.safetensors +3 -0
  41. collei_000009500.safetensors +3 -0
  42. collei_000009750.safetensors +3 -0
  43. config.yaml +99 -0
  44. log.txt +0 -0
  45. logs/0_log.txt +233 -0
  46. logs/1_log.txt +233 -0
  47. logs/2_log.txt +233 -0
  48. optimizer.pt +3 -0
  49. samples/1754471571065__000000000_0.webp +3 -0
  50. samples/1754471955993__000000250_0.webp +3 -0
.gitattributes CHANGED
@@ -33,3 +33,44 @@ 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/1754471571065__000000000_0.webp filter=lfs diff=lfs merge=lfs -text
37
+ samples/1754471955993__000000250_0.webp filter=lfs diff=lfs merge=lfs -text
38
+ samples/1754472337285__000000500_0.webp filter=lfs diff=lfs merge=lfs -text
39
+ samples/1754472716595__000000750_0.webp filter=lfs diff=lfs merge=lfs -text
40
+ samples/1754473097548__000001000_0.webp filter=lfs diff=lfs merge=lfs -text
41
+ samples/1754473484249__000001250_0.webp filter=lfs diff=lfs merge=lfs -text
42
+ samples/1754473871102__000001500_0.webp filter=lfs diff=lfs merge=lfs -text
43
+ samples/1754474252606__000001750_0.webp filter=lfs diff=lfs merge=lfs -text
44
+ samples/1754474631339__000002000_0.webp filter=lfs diff=lfs merge=lfs -text
45
+ samples/1754475020210__000002250_0.webp filter=lfs diff=lfs merge=lfs -text
46
+ samples/1754475400800__000002500_0.webp filter=lfs diff=lfs merge=lfs -text
47
+ samples/1754475781198__000002750_0.webp filter=lfs diff=lfs merge=lfs -text
48
+ samples/1754476158933__000003000_0.webp filter=lfs diff=lfs merge=lfs -text
49
+ samples/1754476540291__000003250_0.webp filter=lfs diff=lfs merge=lfs -text
50
+ samples/1754476916403__000003500_0.webp filter=lfs diff=lfs merge=lfs -text
51
+ samples/1754477294553__000003750_0.webp filter=lfs diff=lfs merge=lfs -text
52
+ samples/1754477666518__000004000_0.webp filter=lfs diff=lfs merge=lfs -text
53
+ samples/1754478051599__000004250_0.webp filter=lfs diff=lfs merge=lfs -text
54
+ samples/1754478426966__000004500_0.webp filter=lfs diff=lfs merge=lfs -text
55
+ samples/1754478803190__000004750_0.webp filter=lfs diff=lfs merge=lfs -text
56
+ samples/1754479182097__000005000_0.webp filter=lfs diff=lfs merge=lfs -text
57
+ samples/1754479562281__000005250_0.webp filter=lfs diff=lfs merge=lfs -text
58
+ samples/1754479940051__000005500_0.webp filter=lfs diff=lfs merge=lfs -text
59
+ samples/1754480318651__000005750_0.webp filter=lfs diff=lfs merge=lfs -text
60
+ samples/1754480694982__000006000_0.webp filter=lfs diff=lfs merge=lfs -text
61
+ samples/1754481079325__000006250_0.webp filter=lfs diff=lfs merge=lfs -text
62
+ samples/1754481458765__000006500_0.webp filter=lfs diff=lfs merge=lfs -text
63
+ samples/1754481834152__000006750_0.webp filter=lfs diff=lfs merge=lfs -text
64
+ samples/1754482209861__000007000_0.webp filter=lfs diff=lfs merge=lfs -text
65
+ samples/1754482594376__000007250_0.webp filter=lfs diff=lfs merge=lfs -text
66
+ samples/1754482971660__000007500_0.webp filter=lfs diff=lfs merge=lfs -text
67
+ samples/1754483352744__000007750_0.webp filter=lfs diff=lfs merge=lfs -text
68
+ samples/1754483732176__000008000_0.webp filter=lfs diff=lfs merge=lfs -text
69
+ samples/1754484117340__000008250_0.webp filter=lfs diff=lfs merge=lfs -text
70
+ samples/1754484492999__000008500_0.webp filter=lfs diff=lfs merge=lfs -text
71
+ samples/1754484873785__000008750_0.webp filter=lfs diff=lfs merge=lfs -text
72
+ samples/1754485252837__000009000_0.webp filter=lfs diff=lfs merge=lfs -text
73
+ samples/1754485642263__000009250_0.webp filter=lfs diff=lfs merge=lfs -text
74
+ samples/1754486027665__000009500_0.webp filter=lfs diff=lfs merge=lfs -text
75
+ samples/1754486406571__000009750_0.webp filter=lfs diff=lfs merge=lfs -text
76
+ samples/1754486839983__000010000_0.webp filter=lfs diff=lfs merge=lfs -text
.job_config.json ADDED
@@ -0,0 +1,116 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "job": "extension",
3
+ "config": {
4
+ "name": "collei",
5
+ "process": [
6
+ {
7
+ "type": "ui_trainer",
8
+ "training_folder": "/home/featurize/ai-toolkit/output",
9
+ "sqlite_db_path": "/home/featurize/ai-toolkit/aitk_db.db",
10
+ "device": "cuda",
11
+ "trigger_word": null,
12
+ "performance_log_every": 10,
13
+ "network": {
14
+ "type": "lora",
15
+ "linear": 32,
16
+ "linear_alpha": 32,
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": 250,
28
+ "max_step_saves_to_keep": 4000,
29
+ "save_format": "diffusers",
30
+ "push_to_hub": false
31
+ },
32
+ "datasets": [
33
+ {
34
+ "folder_path": "/home/featurize/ai-toolkit/datasets/genshin_impact_COLLEI_images_and_texts",
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
+ 256
47
+ ],
48
+ "controls": [],
49
+ "shrink_video_to_frames": true,
50
+ "num_frames": 1
51
+ }
52
+ ],
53
+ "train": {
54
+ "batch_size": 1,
55
+ "bypass_guidance_embedding": false,
56
+ "steps": 10000,
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": "weighted",
64
+ "content_or_style": "balanced",
65
+ "optimizer_params": {
66
+ "weight_decay": 0.0001
67
+ },
68
+ "unload_text_encoder": false,
69
+ "lr": 0.0001,
70
+ "ema_config": {
71
+ "use_ema": false,
72
+ "ema_decay": 0.99
73
+ },
74
+ "skip_first_sample": false,
75
+ "disable_sampling": false,
76
+ "dtype": "bf16",
77
+ "diff_output_preservation": false,
78
+ "diff_output_preservation_multiplier": 1,
79
+ "diff_output_preservation_class": "person"
80
+ },
81
+ "model": {
82
+ "name_or_path": "Wan-AI/Wan2.2-TI2V-5B-Diffusers",
83
+ "quantize": true,
84
+ "qtype": "qfloat8",
85
+ "quantize_te": true,
86
+ "qtype_te": "qfloat8",
87
+ "arch": "wan22_5b",
88
+ "low_vram": true,
89
+ "model_kwargs": {}
90
+ },
91
+ "sample": {
92
+ "sampler": "flowmatch",
93
+ "sample_every": 250,
94
+ "width": 512,
95
+ "height": 512,
96
+ "samples": [
97
+ {
98
+ "prompt": "In this digital anime-style drawing, COLLEI, with flowing green hair and purple eyes, sits on a bed holding a green gift box adorned with a yellow ribbon. They wear a green and gold outfit with intricate designs. The background features a softly lit room with a wooden dresser and a vase of flowers. The overall atmosphere is cozy and intimate, with a focus on the character's thoughtful expression."
99
+ }
100
+ ],
101
+ "neg": "",
102
+ "seed": 42,
103
+ "walk_seed": true,
104
+ "guidance_scale": 4,
105
+ "sample_steps": 25,
106
+ "num_frames": 121,
107
+ "fps": 24
108
+ }
109
+ }
110
+ ]
111
+ },
112
+ "meta": {
113
+ "name": "[name]",
114
+ "version": "1.0"
115
+ }
116
+ }
collei.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8a8539c5d1553d17a4417078acc3df564fb518361fac849f183694904000d7bc
3
+ size 161293128
collei_000000250.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1428cfdac1c8da0eb92044ac07176dc82dce5eab52434bdde3d0520ed6b8bfb5
3
+ size 161293128
collei_000000500.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2a720ea2ac28bd0b54698471bf4c654d4932730ce36d09ac0be338517012dbf1
3
+ size 161293128
collei_000000750.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:62906435ae6702c7bde59987d7f3d0aa016de74ce896f2640cb622a85ac976d8
3
+ size 161293128
collei_000001000.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4d7380afdcbeb44ed3aa90f132a15fba8e29d1bbf376bc222e30e25384d1f80d
3
+ size 161293128
collei_000001250.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e670f471e96f26707dd05d1a8c3500a4560cd84a481f886040e6b0a909db7968
3
+ size 161293128
collei_000001500.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3691c31c86a9d8da36eec90ec032b345d6e511ba010dc23ac7faa384abca8107
3
+ size 161293128
collei_000001750.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7bc9048c56ba41f9d2c44d20386b42c5cdcd0ee7b9a192f2c4ba797437081a43
3
+ size 161293128
collei_000002000.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d600dd1a866869b2f7b5c7265065a41c7db0594ff79616a2287c892528308f03
3
+ size 161293128
collei_000002250.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1489b68ec306adffcfe4fe13869f29f83d2e903e49fd8b98059e3c5dd1675610
3
+ size 161293128
collei_000002500.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dccf4aba9d7c69d063d634c263890d9cd19f38a8f2b9ba82ea109c177243a7bc
3
+ size 161293128
collei_000002750.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:11ff5bf39b7b726d4e520000a2570d9f641fa42a00b89621cdbc3905997ab6c0
3
+ size 161293128
collei_000003000.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:64982be31ee843942cd676eea2bb2e9cbb8b178a757a1806293d1105083afc00
3
+ size 161293128
collei_000003250.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5fc2284a8454b292820ab48bbf7294b2b083aee8bc048c6bbc057a8f53a24044
3
+ size 161293128
collei_000003500.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3c0d57115ea3d56fd5500b41ab36926420c10f23ffde2ac85601b242a9478cf2
3
+ size 161293128
collei_000003750.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a4dd9c00ac9ef22c97d6fb4a92f292240bc157748f63db76e49371b63ff12e29
3
+ size 161293128
collei_000004000.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:02e4a893c3c4ad6b6c5272b4b8ff081f2e267fc15c958f13eed639eb11d47806
3
+ size 161293128
collei_000004250.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b34d6222b0f101aea582f2d7bf205f4f7f8fbd8499a51bf34ab45b62cbb4e003
3
+ size 161293128
collei_000004500.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8dc9d28b94b6e7d1bb8cdd12b1ce27baf0b2e1f0659fdfaa0ef426f1aa2e85be
3
+ size 161293128
collei_000004750.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ac6c7a935e42bf2b5ffdc699d3329392cc1f91a529cd6541537cc6e75d94a6e3
3
+ size 161293128
collei_000005000.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1c8b82a40ac404d9b237d6fa57d0fab9a42e6de72238308cf0cceb6e31be6705
3
+ size 161293128
collei_000005250.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:84ae10b7361a91213a03520b2fa65714270c25b00c1480228c1b5a4d2cd0e47b
3
+ size 161293128
collei_000005500.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b862edbcd3ab7a9c35fdb5133186f1d6ecb6d67553cc8b24e4796acc5489065d
3
+ size 161293128
collei_000005750.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1bd26f0fb0ecbcc62f19a02463a93c93d5d8b3956d7f4e41b0ccfca259b0bbfa
3
+ size 161293128
collei_000006000.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:12a76875c58af2578c4b54f3da3abf8d832872f2cb670585a5436112ab2810f7
3
+ size 161293128
collei_000006250.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cbaccca86e0f252cbc8a4e735becdac8dbde259decb2d5a2681fa3d4b7ef919f
3
+ size 161293128
collei_000006500.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c1eb35b64bf87ac07b6e5487e939f05f796b283e9002e557d76ab75e9e698b40
3
+ size 161293128
collei_000006750.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:514bb48edda2f7bad528173cdffd4ded51a5d03d1fea7f4e3302718f2f10ad20
3
+ size 161293128
collei_000007000.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e1009bd11d3b5b51de6a5eba80e50eaaf464614525a7c53f7e7b8d22d4320365
3
+ size 161293128
collei_000007250.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:af92a57e241007f2fb48abfd054ea873c80460a891534b9d7f1c13bbf7b90ad1
3
+ size 161293128
collei_000007500.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a973eff77d952aa633dc6a53508fc17d3a12f6cbb01e30171edaef6e0adc4ba5
3
+ size 161293128
collei_000007750.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:195e462ff471b85cc3b76bac19e1e57ef223559afb145436b42b5ac7bc91953e
3
+ size 161293128
collei_000008000.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2e844926d10690f7ceb92fa60a7e03c382450aa40eeff29cb2b3b738b3c706d8
3
+ size 161293128
collei_000008250.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2330e4f12df635d79f35f8ee9439410642fa633cc5187791f36f3cce44af625e
3
+ size 161293128
collei_000008500.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0d5c74680f5c9a7d15dd8af5a3df25b805321d37f8169f351e40282411472be8
3
+ size 161293128
collei_000008750.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6f74dd01d213b758dd4384d68656209a06af01a86fe6c486a52819c8d62dc2c8
3
+ size 161293128
collei_000009000.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f1c38981e2820648c66e23c696847a4e4d7989e43d5b40c8a723daba0b4d2003
3
+ size 161293128
collei_000009250.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:44dceb5f345a0a6dbe646a8393615bff017d1440f393c23ea778567f05cf0bb9
3
+ size 161293128
collei_000009500.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9c0f1a42247b71f91ddf4e718f707566b007b0dfaa4dbf257a2dfa40577ac2dd
3
+ size 161293128
collei_000009750.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:94fe6411fafa5425819a470040acdf3b15a777a23534b195006e97bd34ba29ed
3
+ size 161293128
config.yaml ADDED
@@ -0,0 +1,99 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ job: extension
2
+ config:
3
+ name: collei
4
+ process:
5
+ - type: ui_trainer
6
+ training_folder: /home/featurize/ai-toolkit/output
7
+ sqlite_db_path: /home/featurize/ai-toolkit/aitk_db.db
8
+ device: cuda
9
+ trigger_word: null
10
+ performance_log_every: 10
11
+ network:
12
+ type: lora
13
+ linear: 32
14
+ linear_alpha: 32
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: 250
24
+ max_step_saves_to_keep: 4000
25
+ save_format: diffusers
26
+ push_to_hub: false
27
+ datasets:
28
+ - folder_path: /home/featurize/ai-toolkit/datasets/genshin_impact_COLLEI_images_and_texts
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
+ - 256
41
+ controls: []
42
+ shrink_video_to_frames: true
43
+ num_frames: 1
44
+ train:
45
+ batch_size: 1
46
+ bypass_guidance_embedding: false
47
+ steps: 10000
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: weighted
55
+ content_or_style: balanced
56
+ optimizer_params:
57
+ weight_decay: 0.0001
58
+ unload_text_encoder: false
59
+ lr: 0.0001
60
+ ema_config:
61
+ use_ema: false
62
+ ema_decay: 0.99
63
+ skip_first_sample: false
64
+ disable_sampling: false
65
+ dtype: bf16
66
+ diff_output_preservation: false
67
+ diff_output_preservation_multiplier: 1
68
+ diff_output_preservation_class: person
69
+ model:
70
+ name_or_path: Wan-AI/Wan2.2-TI2V-5B-Diffusers
71
+ quantize: true
72
+ qtype: qfloat8
73
+ quantize_te: true
74
+ qtype_te: qfloat8
75
+ arch: wan22_5b
76
+ low_vram: true
77
+ model_kwargs: {}
78
+ sample:
79
+ sampler: flowmatch
80
+ sample_every: 250
81
+ width: 512
82
+ height: 512
83
+ samples:
84
+ - prompt: In this digital anime-style drawing, COLLEI, with flowing green hair
85
+ and purple eyes, sits on a bed holding a green gift box adorned with a yellow
86
+ ribbon. They wear a green and gold outfit with intricate designs. The background
87
+ features a softly lit room with a wooden dresser and a vase of flowers.
88
+ The overall atmosphere is cozy and intimate, with a focus on the character's
89
+ thoughtful expression.
90
+ neg: ''
91
+ seed: 42
92
+ walk_seed: true
93
+ guidance_scale: 4
94
+ sample_steps: 25
95
+ num_frames: 121
96
+ fps: 24
97
+ meta:
98
+ name: collei
99
+ 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,233 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Running 1 job
2
+ {
3
+ "type": "ui_trainer",
4
+ "training_folder": "/home/featurize/ai-toolkit/output",
5
+ "sqlite_db_path": "/home/featurize/ai-toolkit/aitk_db.db",
6
+ "device": "cuda",
7
+ "trigger_word": null,
8
+ "performance_log_every": 10,
9
+ "network": {
10
+ "type": "lora",
11
+ "linear": 32,
12
+ "linear_alpha": 32,
13
+ "conv": 16,
14
+ "conv_alpha": 16,
15
+ "lokr_full_rank": true,
16
+ "lokr_factor": -1,
17
+ "network_kwargs": {
18
+ "ignore_if_contains": []
19
+ }
20
+ },
21
+ "save": {
22
+ "dtype": "bf16",
23
+ "save_every": 250,
24
+ "max_step_saves_to_keep": 4000,
25
+ "save_format": "diffusers",
26
+ "push_to_hub": false
27
+ },
28
+ "datasets": [
29
+ {
30
+ "folder_path": "/home/featurize/ai-toolkit/datasets/genshin_impact_COLLEI_images_and_texts",
31
+ "control_path": null,
32
+ "mask_path": null,
33
+ "mask_min_value": 0.1,
34
+ "default_caption": "",
35
+ "caption_ext": "txt",
36
+ "caption_dropout_rate": 0.05,
37
+ "cache_latents_to_disk": false,
38
+ "is_reg": false,
39
+ "network_weight": 1,
40
+ "resolution": [
41
+ 512,
42
+ 256
43
+ ],
44
+ "controls": [],
45
+ "shrink_video_to_frames": true,
46
+ "num_frames": 1
47
+ }
48
+ ],
49
+ "train": {
50
+ "batch_size": 1,
51
+ "bypass_guidance_embedding": false,
52
+ "steps": 10000,
53
+ "gradient_accumulation": 1,
54
+ "train_unet": true,
55
+ "train_text_encoder": false,
56
+ "gradient_checkpointing": true,
57
+ "noise_scheduler": "flowmatch",
58
+ "optimizer": "adamw8bit",
59
+ "timestep_type": "weighted",
60
+ "content_or_style": "balanced",
61
+ "optimizer_params": {
62
+ "weight_decay": 0.0001
63
+ },
64
+ "unload_text_encoder": false,
65
+ "lr": 0.0001,
66
+ "ema_config": {
67
+ "use_ema": false,
68
+ "ema_decay": 0.99
69
+ },
70
+ "skip_first_sample": false,
71
+ "disable_sampling": false,
72
+ "dtype": "bf16",
73
+ "diff_output_preservation": false,
74
+ "diff_output_preservation_multiplier": 1,
75
+ "diff_output_preservation_class": "person"
76
+ },
77
+ "model": {
78
+ "name_or_path": "Wan-AI/Wan2.2-TI2V-5B-Diffusers",
79
+ "quantize": true,
80
+ "qtype": "qfloat8",
81
+ "quantize_te": true,
82
+ "qtype_te": "qfloat8",
83
+ "arch": "wan22_5b",
84
+ "low_vram": true,
85
+ "model_kwargs": {}
86
+ },
87
+ "sample": {
88
+ "sampler": "flowmatch",
89
+ "sample_every": 250,
90
+ "width": 512,
91
+ "height": 512,
92
+ "samples": [
93
+ {
94
+ "prompt": "In this digital anime-style drawing, COLLEI, with flowing green hair and purple eyes, sits on a bed holding a green gift box adorned with a yellow ribbon. They wear a green and gold outfit with intricate designs. The background features a softly lit room with a wooden dresser and a vase of flowers. The overall atmosphere is cozy and intimate, with a focus on the character's thoughtful expression."
95
+ }
96
+ ],
97
+ "neg": "",
98
+ "seed": 42,
99
+ "walk_seed": true,
100
+ "guidance_scale": 4,
101
+ "sample_steps": 25,
102
+ "num_frames": 121,
103
+ "fps": 24
104
+ }
105
+ }
106
+ Using EMA
107
+ Changing sample extention to animated webp
108
+ Using SQLite database at /home/featurize/ai-toolkit/aitk_db.db
109
+ Job ID: "cc24dbd1-fe41-4602-b95c-6e9677f80030"
110
+
111
+ #############################################
112
+ # Running job: collei
113
+ #############################################
114
+
115
+
116
+ Running 1 process
117
+ Loading Wan model
118
+ Loading transformer
119
+
120
+
121
+ Error running job: An error occurred while downloading using `hf_transfer`. Consider disabling HF_HUB_ENABLE_HF_TRANSFER for better error handling.
122
+
123
+ ========================================
124
+ Result:
125
+ - 0 completed jobs
126
+ - 1 failure
127
+ ========================================
128
+ Traceback (most recent call last):
129
+ Traceback (most recent call last):
130
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 470, in http_get
131
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 470, in http_get
132
+ hf_transfer.download(hf_transfer.download(
133
+
134
+ ExceptionException: : Failed too many failures in parallel (3): Request: error decoding response body (no permits available)Failed too many failures in parallel (3): Request: error decoding response body (no permits available)
135
+
136
+
137
+
138
+ The above exception was the direct cause of the following exception:
139
+ The above exception was the direct cause of the following exception:
140
+
141
+
142
+ Traceback (most recent call last):
143
+ Traceback (most recent call last):
144
+ File "/home/featurize/ai-toolkit/run.py", line 120, in <module>
145
+ File "/home/featurize/ai-toolkit/run.py", line 120, in <module>
146
+ main()main()
147
+
148
+ File "/home/featurize/ai-toolkit/run.py", line 108, in main
149
+ File "/home/featurize/ai-toolkit/run.py", line 108, in main
150
+ raise eraise e
151
+
152
+ File "/home/featurize/ai-toolkit/run.py", line 96, in main
153
+ File "/home/featurize/ai-toolkit/run.py", line 96, in main
154
+ job.run()job.run()
155
+
156
+ File "/home/featurize/ai-toolkit/jobs/ExtensionJob.py", line 22, in run
157
+ File "/home/featurize/ai-toolkit/jobs/ExtensionJob.py", line 22, in run
158
+ process.run()process.run()
159
+
160
+ File "/home/featurize/ai-toolkit/jobs/process/BaseSDTrainProcess.py", line 1518, in run
161
+ File "/home/featurize/ai-toolkit/jobs/process/BaseSDTrainProcess.py", line 1518, in run
162
+ self.sd.load_model()self.sd.load_model()
163
+
164
+ File "/home/featurize/ai-toolkit/extensions_built_in/diffusion_models/wan22/wan22_model.py", line 109, in load_model
165
+ File "/home/featurize/ai-toolkit/extensions_built_in/diffusion_models/wan22/wan22_model.py", line 109, in load_model
166
+ super().load_model()super().load_model()
167
+
168
+ File "/home/featurize/ai-toolkit/toolkit/models/wan21/wan21.py", line 358, in load_model
169
+ File "/home/featurize/ai-toolkit/toolkit/models/wan21/wan21.py", line 358, in load_model
170
+ transformer = WanTransformer3DModel.from_pretrained(transformer = WanTransformer3DModel.from_pretrained(
171
+
172
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
173
+
174
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
175
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
176
+ return fn(*args, **kwargs)return fn(*args, **kwargs)
177
+
178
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
179
+
180
+ File "/environment/miniconda3/lib/python3.11/site-packages/diffusers/models/modeling_utils.py", line 1202, in from_pretrained
181
+ File "/environment/miniconda3/lib/python3.11/site-packages/diffusers/models/modeling_utils.py", line 1202, in from_pretrained
182
+ resolved_model_file, sharded_metadata = _get_checkpoint_shard_files(resolved_model_file, sharded_metadata = _get_checkpoint_shard_files(
183
+
184
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
185
+
186
+ File "/environment/miniconda3/lib/python3.11/site-packages/diffusers/utils/hub_utils.py", line 417, in _get_checkpoint_shard_files
187
+ File "/environment/miniconda3/lib/python3.11/site-packages/diffusers/utils/hub_utils.py", line 417, in _get_checkpoint_shard_files
188
+ cached_folder = snapshot_download(cached_folder = snapshot_download(
189
+
190
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
191
+
192
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
193
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
194
+ return fn(*args, **kwargs)return fn(*args, **kwargs)
195
+
196
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
197
+
198
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/_snapshot_download.py", line 330, in snapshot_download
199
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/_snapshot_download.py", line 330, in snapshot_download
200
+ _inner_hf_hub_download(file)_inner_hf_hub_download(file)
201
+
202
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/_snapshot_download.py", line 306, in _inner_hf_hub_download
203
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/_snapshot_download.py", line 306, in _inner_hf_hub_download
204
+ return hf_hub_download(return hf_hub_download(
205
+
206
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
207
+
208
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
209
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
210
+ return fn(*args, **kwargs)return fn(*args, **kwargs)
211
+
212
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
213
+
214
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 1010, in hf_hub_download
215
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 1010, in hf_hub_download
216
+ return _hf_hub_download_to_cache_dir(return _hf_hub_download_to_cache_dir(
217
+
218
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
219
+
220
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 1171, in _hf_hub_download_to_cache_dir
221
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 1171, in _hf_hub_download_to_cache_dir
222
+ _download_to_tmp_and_move(_download_to_tmp_and_move(
223
+
224
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 1738, in _download_to_tmp_and_move
225
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 1738, in _download_to_tmp_and_move
226
+ http_get(http_get(
227
+
228
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 481, in http_get
229
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 481, in http_get
230
+ raise RuntimeError(raise RuntimeError(
231
+
232
+ RuntimeErrorRuntimeError: : An error occurred while downloading using `hf_transfer`. Consider disabling HF_HUB_ENABLE_HF_TRANSFER for better error handling.An error occurred while downloading using `hf_transfer`. Consider disabling HF_HUB_ENABLE_HF_TRANSFER for better error handling.
233
+
logs/1_log.txt ADDED
@@ -0,0 +1,233 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Running 1 job
2
+ {
3
+ "type": "ui_trainer",
4
+ "training_folder": "/home/featurize/ai-toolkit/output",
5
+ "sqlite_db_path": "/home/featurize/ai-toolkit/aitk_db.db",
6
+ "device": "cuda",
7
+ "trigger_word": null,
8
+ "performance_log_every": 10,
9
+ "network": {
10
+ "type": "lora",
11
+ "linear": 32,
12
+ "linear_alpha": 32,
13
+ "conv": 16,
14
+ "conv_alpha": 16,
15
+ "lokr_full_rank": true,
16
+ "lokr_factor": -1,
17
+ "network_kwargs": {
18
+ "ignore_if_contains": []
19
+ }
20
+ },
21
+ "save": {
22
+ "dtype": "bf16",
23
+ "save_every": 250,
24
+ "max_step_saves_to_keep": 4000,
25
+ "save_format": "diffusers",
26
+ "push_to_hub": false
27
+ },
28
+ "datasets": [
29
+ {
30
+ "folder_path": "/home/featurize/ai-toolkit/datasets/genshin_impact_COLLEI_images_and_texts",
31
+ "control_path": null,
32
+ "mask_path": null,
33
+ "mask_min_value": 0.1,
34
+ "default_caption": "",
35
+ "caption_ext": "txt",
36
+ "caption_dropout_rate": 0.05,
37
+ "cache_latents_to_disk": false,
38
+ "is_reg": false,
39
+ "network_weight": 1,
40
+ "resolution": [
41
+ 512,
42
+ 256
43
+ ],
44
+ "controls": [],
45
+ "shrink_video_to_frames": true,
46
+ "num_frames": 1
47
+ }
48
+ ],
49
+ "train": {
50
+ "batch_size": 1,
51
+ "bypass_guidance_embedding": false,
52
+ "steps": 10000,
53
+ "gradient_accumulation": 1,
54
+ "train_unet": true,
55
+ "train_text_encoder": false,
56
+ "gradient_checkpointing": true,
57
+ "noise_scheduler": "flowmatch",
58
+ "optimizer": "adamw8bit",
59
+ "timestep_type": "weighted",
60
+ "content_or_style": "balanced",
61
+ "optimizer_params": {
62
+ "weight_decay": 0.0001
63
+ },
64
+ "unload_text_encoder": false,
65
+ "lr": 0.0001,
66
+ "ema_config": {
67
+ "use_ema": false,
68
+ "ema_decay": 0.99
69
+ },
70
+ "skip_first_sample": false,
71
+ "disable_sampling": false,
72
+ "dtype": "bf16",
73
+ "diff_output_preservation": false,
74
+ "diff_output_preservation_multiplier": 1,
75
+ "diff_output_preservation_class": "person"
76
+ },
77
+ "model": {
78
+ "name_or_path": "Wan-AI/Wan2.2-TI2V-5B-Diffusers",
79
+ "quantize": true,
80
+ "qtype": "qfloat8",
81
+ "quantize_te": true,
82
+ "qtype_te": "qfloat8",
83
+ "arch": "wan22_5b",
84
+ "low_vram": true,
85
+ "model_kwargs": {}
86
+ },
87
+ "sample": {
88
+ "sampler": "flowmatch",
89
+ "sample_every": 250,
90
+ "width": 512,
91
+ "height": 512,
92
+ "samples": [
93
+ {
94
+ "prompt": "In this digital anime-style drawing, COLLEI, with flowing green hair and purple eyes, sits on a bed holding a green gift box adorned with a yellow ribbon. They wear a green and gold outfit with intricate designs. The background features a softly lit room with a wooden dresser and a vase of flowers. The overall atmosphere is cozy and intimate, with a focus on the character's thoughtful expression."
95
+ }
96
+ ],
97
+ "neg": "",
98
+ "seed": 42,
99
+ "walk_seed": true,
100
+ "guidance_scale": 4,
101
+ "sample_steps": 25,
102
+ "num_frames": 121,
103
+ "fps": 24
104
+ }
105
+ }
106
+ Using EMA
107
+ Changing sample extention to animated webp
108
+ Using SQLite database at /home/featurize/ai-toolkit/aitk_db.db
109
+ Job ID: "cc24dbd1-fe41-4602-b95c-6e9677f80030"
110
+
111
+ #############################################
112
+ # Running job: collei
113
+ #############################################
114
+
115
+
116
+ Running 1 process
117
+ Loading Wan model
118
+ Loading transformer
119
+ Error running job: Fast download using 'hf_transfer' is enabled (HF_HUB_ENABLE_HF_TRANSFER=1) but 'hf_transfer' package is not available in your environment. Try `pip install hf_transfer`.
120
+
121
+ ========================================
122
+ Result:
123
+ - 0 completed jobs
124
+ - 1 failure
125
+ ========================================
126
+ Traceback (most recent call last):
127
+ Traceback (most recent call last):
128
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 402, in http_get
129
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 402, in http_get
130
+ import hf_transfer # type: ignore[no-redef]import hf_transfer # type: ignore[no-redef]
131
+
132
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
133
+
134
+ ModuleNotFoundErrorModuleNotFoundError: : No module named 'hf_transfer'No module named 'hf_transfer'
135
+
136
+
137
+
138
+ During handling of the above exception, another exception occurred:
139
+ During handling of the above exception, another exception occurred:
140
+
141
+
142
+ Traceback (most recent call last):
143
+ Traceback (most recent call last):
144
+ File "/home/featurize/ai-toolkit/run.py", line 120, in <module>
145
+ File "/home/featurize/ai-toolkit/run.py", line 120, in <module>
146
+ main()main()
147
+
148
+ File "/home/featurize/ai-toolkit/run.py", line 108, in main
149
+ File "/home/featurize/ai-toolkit/run.py", line 108, in main
150
+ raise eraise e
151
+
152
+ File "/home/featurize/ai-toolkit/run.py", line 96, in main
153
+ File "/home/featurize/ai-toolkit/run.py", line 96, in main
154
+ job.run()job.run()
155
+
156
+ File "/home/featurize/ai-toolkit/jobs/ExtensionJob.py", line 22, in run
157
+ File "/home/featurize/ai-toolkit/jobs/ExtensionJob.py", line 22, in run
158
+ process.run()process.run()
159
+
160
+ File "/home/featurize/ai-toolkit/jobs/process/BaseSDTrainProcess.py", line 1518, in run
161
+ File "/home/featurize/ai-toolkit/jobs/process/BaseSDTrainProcess.py", line 1518, in run
162
+ self.sd.load_model()self.sd.load_model()
163
+
164
+ File "/home/featurize/ai-toolkit/extensions_built_in/diffusion_models/wan22/wan22_model.py", line 109, in load_model
165
+ File "/home/featurize/ai-toolkit/extensions_built_in/diffusion_models/wan22/wan22_model.py", line 109, in load_model
166
+ super().load_model()super().load_model()
167
+
168
+ File "/home/featurize/ai-toolkit/toolkit/models/wan21/wan21.py", line 358, in load_model
169
+ File "/home/featurize/ai-toolkit/toolkit/models/wan21/wan21.py", line 358, in load_model
170
+ transformer = WanTransformer3DModel.from_pretrained(transformer = WanTransformer3DModel.from_pretrained(
171
+
172
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
173
+
174
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
175
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
176
+ return fn(*args, **kwargs)return fn(*args, **kwargs)
177
+
178
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
179
+
180
+ File "/environment/miniconda3/lib/python3.11/site-packages/diffusers/models/modeling_utils.py", line 1202, in from_pretrained
181
+ File "/environment/miniconda3/lib/python3.11/site-packages/diffusers/models/modeling_utils.py", line 1202, in from_pretrained
182
+ resolved_model_file, sharded_metadata = _get_checkpoint_shard_files(resolved_model_file, sharded_metadata = _get_checkpoint_shard_files(
183
+
184
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
185
+
186
+ File "/environment/miniconda3/lib/python3.11/site-packages/diffusers/utils/hub_utils.py", line 417, in _get_checkpoint_shard_files
187
+ File "/environment/miniconda3/lib/python3.11/site-packages/diffusers/utils/hub_utils.py", line 417, in _get_checkpoint_shard_files
188
+ cached_folder = snapshot_download(cached_folder = snapshot_download(
189
+
190
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
191
+
192
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
193
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
194
+ return fn(*args, **kwargs)return fn(*args, **kwargs)
195
+
196
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
197
+
198
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/_snapshot_download.py", line 330, in snapshot_download
199
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/_snapshot_download.py", line 330, in snapshot_download
200
+ _inner_hf_hub_download(file)_inner_hf_hub_download(file)
201
+
202
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/_snapshot_download.py", line 306, in _inner_hf_hub_download
203
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/_snapshot_download.py", line 306, in _inner_hf_hub_download
204
+ return hf_hub_download(return hf_hub_download(
205
+
206
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
207
+
208
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
209
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
210
+ return fn(*args, **kwargs)return fn(*args, **kwargs)
211
+
212
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
213
+
214
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 1010, in hf_hub_download
215
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 1010, in hf_hub_download
216
+ return _hf_hub_download_to_cache_dir(return _hf_hub_download_to_cache_dir(
217
+
218
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
219
+
220
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 1171, in _hf_hub_download_to_cache_dir
221
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 1171, in _hf_hub_download_to_cache_dir
222
+ _download_to_tmp_and_move(_download_to_tmp_and_move(
223
+
224
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 1738, in _download_to_tmp_and_move
225
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 1738, in _download_to_tmp_and_move
226
+ http_get(http_get(
227
+
228
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 404, in http_get
229
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 404, in http_get
230
+ raise ValueError(raise ValueError(
231
+
232
+ ValueErrorValueError: : Fast download using 'hf_transfer' is enabled (HF_HUB_ENABLE_HF_TRANSFER=1) but 'hf_transfer' package is not available in your environment. Try `pip install hf_transfer`.Fast download using 'hf_transfer' is enabled (HF_HUB_ENABLE_HF_TRANSFER=1) but 'hf_transfer' package is not available in your environment. Try `pip install hf_transfer`.
233
+
logs/2_log.txt ADDED
@@ -0,0 +1,233 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Running 1 job
2
+ {
3
+ "type": "ui_trainer",
4
+ "training_folder": "/home/featurize/ai-toolkit/output",
5
+ "sqlite_db_path": "/home/featurize/ai-toolkit/aitk_db.db",
6
+ "device": "cuda",
7
+ "trigger_word": null,
8
+ "performance_log_every": 10,
9
+ "network": {
10
+ "type": "lora",
11
+ "linear": 32,
12
+ "linear_alpha": 32,
13
+ "conv": 16,
14
+ "conv_alpha": 16,
15
+ "lokr_full_rank": true,
16
+ "lokr_factor": -1,
17
+ "network_kwargs": {
18
+ "ignore_if_contains": []
19
+ }
20
+ },
21
+ "save": {
22
+ "dtype": "bf16",
23
+ "save_every": 250,
24
+ "max_step_saves_to_keep": 4000,
25
+ "save_format": "diffusers",
26
+ "push_to_hub": false
27
+ },
28
+ "datasets": [
29
+ {
30
+ "folder_path": "/home/featurize/ai-toolkit/datasets/genshin_impact_COLLEI_images_and_texts",
31
+ "control_path": null,
32
+ "mask_path": null,
33
+ "mask_min_value": 0.1,
34
+ "default_caption": "",
35
+ "caption_ext": "txt",
36
+ "caption_dropout_rate": 0.05,
37
+ "cache_latents_to_disk": false,
38
+ "is_reg": false,
39
+ "network_weight": 1,
40
+ "resolution": [
41
+ 512,
42
+ 256
43
+ ],
44
+ "controls": [],
45
+ "shrink_video_to_frames": true,
46
+ "num_frames": 1
47
+ }
48
+ ],
49
+ "train": {
50
+ "batch_size": 1,
51
+ "bypass_guidance_embedding": false,
52
+ "steps": 10000,
53
+ "gradient_accumulation": 1,
54
+ "train_unet": true,
55
+ "train_text_encoder": false,
56
+ "gradient_checkpointing": true,
57
+ "noise_scheduler": "flowmatch",
58
+ "optimizer": "adamw8bit",
59
+ "timestep_type": "weighted",
60
+ "content_or_style": "balanced",
61
+ "optimizer_params": {
62
+ "weight_decay": 0.0001
63
+ },
64
+ "unload_text_encoder": false,
65
+ "lr": 0.0001,
66
+ "ema_config": {
67
+ "use_ema": false,
68
+ "ema_decay": 0.99
69
+ },
70
+ "skip_first_sample": false,
71
+ "disable_sampling": false,
72
+ "dtype": "bf16",
73
+ "diff_output_preservation": false,
74
+ "diff_output_preservation_multiplier": 1,
75
+ "diff_output_preservation_class": "person"
76
+ },
77
+ "model": {
78
+ "name_or_path": "Wan-AI/Wan2.2-TI2V-5B-Diffusers",
79
+ "quantize": true,
80
+ "qtype": "qfloat8",
81
+ "quantize_te": true,
82
+ "qtype_te": "qfloat8",
83
+ "arch": "wan22_5b",
84
+ "low_vram": true,
85
+ "model_kwargs": {}
86
+ },
87
+ "sample": {
88
+ "sampler": "flowmatch",
89
+ "sample_every": 250,
90
+ "width": 512,
91
+ "height": 512,
92
+ "samples": [
93
+ {
94
+ "prompt": "In this digital anime-style drawing, COLLEI, with flowing green hair and purple eyes, sits on a bed holding a green gift box adorned with a yellow ribbon. They wear a green and gold outfit with intricate designs. The background features a softly lit room with a wooden dresser and a vase of flowers. The overall atmosphere is cozy and intimate, with a focus on the character's thoughtful expression."
95
+ }
96
+ ],
97
+ "neg": "",
98
+ "seed": 42,
99
+ "walk_seed": true,
100
+ "guidance_scale": 4,
101
+ "sample_steps": 25,
102
+ "num_frames": 121,
103
+ "fps": 24
104
+ }
105
+ }
106
+ Using EMA
107
+ Changing sample extention to animated webp
108
+ Using SQLite database at /home/featurize/ai-toolkit/aitk_db.db
109
+ Job ID: "cc24dbd1-fe41-4602-b95c-6e9677f80030"
110
+
111
+ #############################################
112
+ # Running job: collei
113
+ #############################################
114
+
115
+
116
+ Running 1 process
117
+ Loading Wan model
118
+ Loading transformer
119
+ Error running job: Fast download using 'hf_transfer' is enabled (HF_HUB_ENABLE_HF_TRANSFER=1) but 'hf_transfer' package is not available in your environment. Try `pip install hf_transfer`.
120
+
121
+ ========================================
122
+ Result:
123
+ - 0 completed jobs
124
+ - 1 failure
125
+ ========================================
126
+ Traceback (most recent call last):
127
+ Traceback (most recent call last):
128
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 402, in http_get
129
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 402, in http_get
130
+ import hf_transfer # type: ignore[no-redef]import hf_transfer # type: ignore[no-redef]
131
+
132
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
133
+
134
+ ModuleNotFoundErrorModuleNotFoundError: : No module named 'hf_transfer'No module named 'hf_transfer'
135
+
136
+
137
+
138
+ During handling of the above exception, another exception occurred:
139
+ During handling of the above exception, another exception occurred:
140
+
141
+
142
+ Traceback (most recent call last):
143
+ Traceback (most recent call last):
144
+ File "/home/featurize/ai-toolkit/run.py", line 120, in <module>
145
+ File "/home/featurize/ai-toolkit/run.py", line 120, in <module>
146
+ main()main()
147
+
148
+ File "/home/featurize/ai-toolkit/run.py", line 108, in main
149
+ File "/home/featurize/ai-toolkit/run.py", line 108, in main
150
+ raise eraise e
151
+
152
+ File "/home/featurize/ai-toolkit/run.py", line 96, in main
153
+ File "/home/featurize/ai-toolkit/run.py", line 96, in main
154
+ job.run()job.run()
155
+
156
+ File "/home/featurize/ai-toolkit/jobs/ExtensionJob.py", line 22, in run
157
+ File "/home/featurize/ai-toolkit/jobs/ExtensionJob.py", line 22, in run
158
+ process.run()process.run()
159
+
160
+ File "/home/featurize/ai-toolkit/jobs/process/BaseSDTrainProcess.py", line 1518, in run
161
+ File "/home/featurize/ai-toolkit/jobs/process/BaseSDTrainProcess.py", line 1518, in run
162
+ self.sd.load_model()self.sd.load_model()
163
+
164
+ File "/home/featurize/ai-toolkit/extensions_built_in/diffusion_models/wan22/wan22_model.py", line 109, in load_model
165
+ File "/home/featurize/ai-toolkit/extensions_built_in/diffusion_models/wan22/wan22_model.py", line 109, in load_model
166
+ super().load_model()super().load_model()
167
+
168
+ File "/home/featurize/ai-toolkit/toolkit/models/wan21/wan21.py", line 358, in load_model
169
+ File "/home/featurize/ai-toolkit/toolkit/models/wan21/wan21.py", line 358, in load_model
170
+ transformer = WanTransformer3DModel.from_pretrained(transformer = WanTransformer3DModel.from_pretrained(
171
+
172
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
173
+
174
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
175
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
176
+ return fn(*args, **kwargs)return fn(*args, **kwargs)
177
+
178
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
179
+
180
+ File "/environment/miniconda3/lib/python3.11/site-packages/diffusers/models/modeling_utils.py", line 1202, in from_pretrained
181
+ File "/environment/miniconda3/lib/python3.11/site-packages/diffusers/models/modeling_utils.py", line 1202, in from_pretrained
182
+ resolved_model_file, sharded_metadata = _get_checkpoint_shard_files(resolved_model_file, sharded_metadata = _get_checkpoint_shard_files(
183
+
184
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
185
+
186
+ File "/environment/miniconda3/lib/python3.11/site-packages/diffusers/utils/hub_utils.py", line 417, in _get_checkpoint_shard_files
187
+ File "/environment/miniconda3/lib/python3.11/site-packages/diffusers/utils/hub_utils.py", line 417, in _get_checkpoint_shard_files
188
+ cached_folder = snapshot_download(cached_folder = snapshot_download(
189
+
190
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
191
+
192
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
193
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
194
+ return fn(*args, **kwargs)return fn(*args, **kwargs)
195
+
196
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
197
+
198
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/_snapshot_download.py", line 330, in snapshot_download
199
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/_snapshot_download.py", line 330, in snapshot_download
200
+ _inner_hf_hub_download(file)_inner_hf_hub_download(file)
201
+
202
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/_snapshot_download.py", line 306, in _inner_hf_hub_download
203
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/_snapshot_download.py", line 306, in _inner_hf_hub_download
204
+ return hf_hub_download(return hf_hub_download(
205
+
206
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
207
+
208
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
209
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
210
+ return fn(*args, **kwargs)return fn(*args, **kwargs)
211
+
212
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
213
+
214
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 1010, in hf_hub_download
215
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 1010, in hf_hub_download
216
+ return _hf_hub_download_to_cache_dir(return _hf_hub_download_to_cache_dir(
217
+
218
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
219
+
220
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 1171, in _hf_hub_download_to_cache_dir
221
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 1171, in _hf_hub_download_to_cache_dir
222
+ _download_to_tmp_and_move(_download_to_tmp_and_move(
223
+
224
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 1738, in _download_to_tmp_and_move
225
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 1738, in _download_to_tmp_and_move
226
+ http_get(http_get(
227
+
228
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 404, in http_get
229
+ File "/environment/miniconda3/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 404, in http_get
230
+ raise ValueError(raise ValueError(
231
+
232
+ ValueErrorValueError: : Fast download using 'hf_transfer' is enabled (HF_HUB_ENABLE_HF_TRANSFER=1) but 'hf_transfer' package is not available in your environment. Try `pip install hf_transfer`.Fast download using 'hf_transfer' is enabled (HF_HUB_ENABLE_HF_TRANSFER=1) but 'hf_transfer' package is not available in your environment. Try `pip install hf_transfer`.
233
+
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dabb64eceb80c71af46e3a8f466940fc84fb1bdfc5e3ea9cd0d024d80bceb233
3
+ size 164444101
samples/1754471571065__000000000_0.webp ADDED

Git LFS Details

  • SHA256: 3785fb56d383d7f0ae72b5aac4fa7048313215aafde48018043eb62316e03c39
  • Pointer size: 132 Bytes
  • Size of remote file: 4.84 MB
samples/1754471955993__000000250_0.webp ADDED

Git LFS Details

  • SHA256: 55f7c72b905af31ed0bbac620fc82412bf6c3325e69eb0b548161a4b521c0dd3
  • Pointer size: 132 Bytes
  • Size of remote file: 4.4 MB