napalm commited on
Commit
da09065
·
verified ·
1 Parent(s): ec4f9b7

Upload 11 files

Browse files
.gitattributes CHANGED
@@ -70,3 +70,4 @@ krea2_selena2/loss_log.db filter=lfs diff=lfs merge=lfs -text
70
  krea2_volk2/loss_log.db filter=lfs diff=lfs merge=lfs -text
71
  krea2_yoh/loss_log.db filter=lfs diff=lfs merge=lfs -text
72
  krea2_rihanna/loss_log.db filter=lfs diff=lfs merge=lfs -text
 
 
70
  krea2_volk2/loss_log.db filter=lfs diff=lfs merge=lfs -text
71
  krea2_yoh/loss_log.db filter=lfs diff=lfs merge=lfs -text
72
  krea2_rihanna/loss_log.db filter=lfs diff=lfs merge=lfs -text
73
+ krea2_ljames3/loss_log.db filter=lfs diff=lfs merge=lfs -text
krea2_ljames3/.job_config.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "job": "extension",
3
+ "config": {
4
+ "name": "krea2_ljames3",
5
+ "process": [
6
+ {
7
+ "type": "diffusion_trainer",
8
+ "training_folder": "/home/sam/ai-toolkit/output",
9
+ "sqlite_db_path": "/home/sam/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
+ "lokr_full_rank": true,
18
+ "lokr_factor": -1,
19
+ "dropout": 0.05,
20
+ "network_kwargs": {
21
+ "ignore_if_contains": []
22
+ }
23
+ },
24
+ "save": {
25
+ "dtype": "bf16",
26
+ "save_every": 250,
27
+ "max_step_saves_to_keep": 4,
28
+ "save_format": "diffusers",
29
+ "push_to_hub": false
30
+ },
31
+ "datasets": [
32
+ {
33
+ "folder_path": "/home/sam/ai-toolkit/datasets/ljames",
34
+ "mask_path": null,
35
+ "mask_min_value": 0.1,
36
+ "default_caption": "",
37
+ "caption_ext": "txt",
38
+ "caption_dropout_rate": 0.05,
39
+ "cache_latents_to_disk": false,
40
+ "is_reg": false,
41
+ "network_weight": 1,
42
+ "resolution": [
43
+ 512,
44
+ 768,
45
+ 1024
46
+ ],
47
+ "controls": [],
48
+ "shrink_video_to_frames": true,
49
+ "num_frames": 1,
50
+ "flip_x": false,
51
+ "flip_y": false,
52
+ "num_repeats": 1
53
+ }
54
+ ],
55
+ "train": {
56
+ "batch_size": 1,
57
+ "bypass_guidance_embedding": false,
58
+ "steps": 6000,
59
+ "gradient_accumulation": 1,
60
+ "train_unet": true,
61
+ "train_text_encoder": false,
62
+ "gradient_checkpointing": true,
63
+ "noise_scheduler": "flowmatch",
64
+ "optimizer": "adamw8bit",
65
+ "timestep_type": "linear",
66
+ "content_or_style": "balanced",
67
+ "optimizer_params": {
68
+ "weight_decay": 0.0001
69
+ },
70
+ "unload_text_encoder": true,
71
+ "cache_text_embeddings": false,
72
+ "lr": 0.0001,
73
+ "ema_config": {
74
+ "use_ema": false,
75
+ "ema_decay": 0.99
76
+ },
77
+ "skip_first_sample": false,
78
+ "force_first_sample": false,
79
+ "disable_sampling": true,
80
+ "dtype": "bf16",
81
+ "diff_output_preservation": false,
82
+ "diff_output_preservation_multiplier": 1,
83
+ "diff_output_preservation_class": "person",
84
+ "switch_boundary_every": 1,
85
+ "loss_type": "mse"
86
+ },
87
+ "logging": {
88
+ "log_every": 1,
89
+ "use_ui_logger": true
90
+ },
91
+ "model": {
92
+ "name_or_path": "/home/sam/ai-toolkit/models/krea2",
93
+ "quantize": true,
94
+ "qtype": "qfloat8",
95
+ "quantize_te": true,
96
+ "qtype_te": "qfloat8",
97
+ "arch": "krea2",
98
+ "low_vram": true,
99
+ "model_kwargs": {},
100
+ "compile": true,
101
+ "layer_offloading": false,
102
+ "layer_offloading_text_encoder_percent": 1,
103
+ "layer_offloading_transformer_percent": 1,
104
+ "block_compile": true
105
+ },
106
+ "sample": {
107
+ "sampler": "flowmatch",
108
+ "sample_every": 250,
109
+ "width": 1024,
110
+ "height": 1024,
111
+ "samples": [],
112
+ "neg": "",
113
+ "seed": 42,
114
+ "walk_seed": true,
115
+ "guidance_scale": 4,
116
+ "sample_steps": 30,
117
+ "num_frames": 1,
118
+ "fps": 1
119
+ }
120
+ }
121
+ ]
122
+ },
123
+ "meta": {
124
+ "name": "[name]",
125
+ "version": "1.0"
126
+ }
127
+ }
krea2_ljames3/config.yaml ADDED
@@ -0,0 +1,107 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ job: extension
2
+ config:
3
+ name: krea2_ljames3
4
+ process:
5
+ - type: diffusion_trainer
6
+ training_folder: /home/sam/ai-toolkit/output
7
+ sqlite_db_path: /home/sam/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
+ lokr_full_rank: true
16
+ lokr_factor: -1
17
+ dropout: 0.05
18
+ network_kwargs:
19
+ ignore_if_contains: []
20
+ save:
21
+ dtype: bf16
22
+ save_every: 250
23
+ max_step_saves_to_keep: 4
24
+ save_format: diffusers
25
+ push_to_hub: false
26
+ datasets:
27
+ - folder_path: /home/sam/ai-toolkit/datasets/ljames
28
+ mask_path: null
29
+ mask_min_value: 0.1
30
+ default_caption: ''
31
+ caption_ext: txt
32
+ caption_dropout_rate: 0.05
33
+ cache_latents_to_disk: false
34
+ is_reg: false
35
+ network_weight: 1
36
+ resolution:
37
+ - 512
38
+ - 768
39
+ - 1024
40
+ controls: []
41
+ shrink_video_to_frames: true
42
+ num_frames: 1
43
+ flip_x: false
44
+ flip_y: false
45
+ num_repeats: 1
46
+ train:
47
+ batch_size: 1
48
+ bypass_guidance_embedding: false
49
+ steps: 6000
50
+ gradient_accumulation: 1
51
+ train_unet: true
52
+ train_text_encoder: false
53
+ gradient_checkpointing: true
54
+ noise_scheduler: flowmatch
55
+ optimizer: adamw8bit
56
+ timestep_type: linear
57
+ content_or_style: balanced
58
+ optimizer_params:
59
+ weight_decay: 0.0001
60
+ unload_text_encoder: true
61
+ cache_text_embeddings: false
62
+ lr: 0.0001
63
+ ema_config:
64
+ use_ema: false
65
+ ema_decay: 0.99
66
+ skip_first_sample: false
67
+ force_first_sample: false
68
+ disable_sampling: true
69
+ dtype: bf16
70
+ diff_output_preservation: false
71
+ diff_output_preservation_multiplier: 1
72
+ diff_output_preservation_class: person
73
+ switch_boundary_every: 1
74
+ loss_type: mse
75
+ logging:
76
+ log_every: 1
77
+ use_ui_logger: true
78
+ model:
79
+ name_or_path: /home/sam/ai-toolkit/models/krea2
80
+ quantize: true
81
+ qtype: qfloat8
82
+ quantize_te: true
83
+ qtype_te: qfloat8
84
+ arch: krea2
85
+ low_vram: true
86
+ model_kwargs: {}
87
+ compile: true
88
+ layer_offloading: false
89
+ layer_offloading_text_encoder_percent: 1
90
+ layer_offloading_transformer_percent: 1
91
+ block_compile: true
92
+ sample:
93
+ sampler: flowmatch
94
+ sample_every: 250
95
+ width: 1024
96
+ height: 1024
97
+ samples: []
98
+ neg: ''
99
+ seed: 42
100
+ walk_seed: true
101
+ guidance_scale: 4
102
+ sample_steps: 30
103
+ num_frames: 1
104
+ fps: 1
105
+ meta:
106
+ name: krea2_ljames3
107
+ version: '1.0'
krea2_ljames3/krea2_ljames3.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:286f14de23f9b92f4e9f3f979d06bec725d800066d95b9fa9f542dec25556e7e
3
+ size 228587720
krea2_ljames3/krea2_ljames3_000005000.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b2212490e41bd1e31d9b371507d97c5772a2428a915cffb6c47475fd67535e1d
3
+ size 228587720
krea2_ljames3/krea2_ljames3_000005250.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a86941f2991f1879b6843b0ae76201fc9c6a19361bc105ba6132c5856754f33b
3
+ size 228587720
krea2_ljames3/krea2_ljames3_000005500.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2f3fe708c7ff524887450992f01086e2184c403dab950d3915363198ea200b1f
3
+ size 228587720
krea2_ljames3/krea2_ljames3_000005750.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a7729419cde57eac12bfa08389af679a02fe4992647d87b7575900d96ae79658
3
+ size 228587720
krea2_ljames3/log.txt ADDED
The diff for this file is too large to render. See raw diff
 
krea2_ljames3/loss_log.db ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8ec3af86a3baf9da3c54e59fd4cec5ece748cb8a73cb92902c82ba3b92bfca28
3
+ size 1105920
krea2_ljames3/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:595bd0d12f27e2f3788ead65517a47b2faf07fe144bf1fa6cfa2467cd41a396e
3
+ size 232702437
krea2_ljames3/pid.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ 2025826