arhiteector commited on
Commit
45c5fdb
·
verified ·
1 Parent(s): cc9c5c6

upload LR_LORA_WAN_512 checkpoints (ignore sqlite wal/shm)

Browse files
.gitattributes CHANGED
@@ -36,3 +36,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
36
  loss_log.db filter=lfs diff=lfs merge=lfs -text
37
  suka512/loss_log.db filter=lfs diff=lfs merge=lfs -text
38
  LR_LORA_512/loss_log.db filter=lfs diff=lfs merge=lfs -text
 
 
36
  loss_log.db filter=lfs diff=lfs merge=lfs -text
37
  suka512/loss_log.db filter=lfs diff=lfs merge=lfs -text
38
  LR_LORA_512/loss_log.db filter=lfs diff=lfs merge=lfs -text
39
+ LR_LORA_WAN_512/loss_log.db filter=lfs diff=lfs merge=lfs -text
LR_LORA_WAN_512/.job_config.json ADDED
@@ -0,0 +1,123 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "job": "extension",
3
+ "config": {
4
+ "name": "LR_LORA_WAN_512",
5
+ "process": [
6
+ {
7
+ "type": "diffusion_trainer",
8
+ "training_folder": "/workspace/ai-toolkit/output",
9
+ "sqlite_db_path": "/workspace/ai-toolkit/aitk_db.db",
10
+ "device": "cuda",
11
+ "trigger_word": "vmvm",
12
+ "performance_log_every": 10,
13
+ "network": {
14
+ "type": "lora",
15
+ "linear": 512,
16
+ "linear_alpha": 512,
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": 4,
29
+ "save_format": "diffusers",
30
+ "push_to_hub": false
31
+ },
32
+ "datasets": [
33
+ {
34
+ "folder_path": "/workspace/ai-toolkit/datasets/wexrct",
35
+ "mask_path": null,
36
+ "mask_min_value": 0.1,
37
+ "default_caption": "vmvm",
38
+ "caption_ext": "txt",
39
+ "caption_dropout_rate": 0.05,
40
+ "cache_latents_to_disk": false,
41
+ "is_reg": false,
42
+ "network_weight": 1,
43
+ "resolution": [
44
+ 1024
45
+ ],
46
+ "controls": [],
47
+ "shrink_video_to_frames": true,
48
+ "num_frames": 1,
49
+ "flip_x": false,
50
+ "flip_y": false,
51
+ "num_repeats": 2
52
+ }
53
+ ],
54
+ "train": {
55
+ "batch_size": 1,
56
+ "bypass_guidance_embedding": false,
57
+ "steps": 3000,
58
+ "gradient_accumulation": 1,
59
+ "train_unet": true,
60
+ "train_text_encoder": false,
61
+ "gradient_checkpointing": true,
62
+ "noise_scheduler": "flowmatch",
63
+ "optimizer": "adamw8bit",
64
+ "timestep_type": "sigmoid",
65
+ "content_or_style": "balanced",
66
+ "optimizer_params": {
67
+ "weight_decay": 0.0001
68
+ },
69
+ "unload_text_encoder": false,
70
+ "cache_text_embeddings": false,
71
+ "lr": 0.0003,
72
+ "ema_config": {
73
+ "use_ema": false,
74
+ "ema_decay": 0.99
75
+ },
76
+ "skip_first_sample": false,
77
+ "force_first_sample": false,
78
+ "disable_sampling": false,
79
+ "dtype": "bf16",
80
+ "diff_output_preservation": false,
81
+ "diff_output_preservation_multiplier": 1,
82
+ "diff_output_preservation_class": "person",
83
+ "switch_boundary_every": 1,
84
+ "loss_type": "mse",
85
+ "do_differential_guidance": true,
86
+ "differential_guidance_scale": 3
87
+ },
88
+ "logging": {
89
+ "log_every": 1,
90
+ "use_ui_logger": true
91
+ },
92
+ "model": {
93
+ "name_or_path": "Wan-AI/Wan2.1-T2V-14B-Diffusers",
94
+ "quantize": true,
95
+ "qtype": "qfloat8",
96
+ "quantize_te": true,
97
+ "qtype_te": "qfloat8",
98
+ "arch": "wan21:14b",
99
+ "low_vram": false,
100
+ "model_kwargs": {}
101
+ },
102
+ "sample": {
103
+ "sampler": "flowmatch",
104
+ "sample_every": 250,
105
+ "width": 1024,
106
+ "height": 1024,
107
+ "samples": [],
108
+ "neg": "",
109
+ "seed": 42,
110
+ "walk_seed": true,
111
+ "guidance_scale": 4,
112
+ "sample_steps": 25,
113
+ "num_frames": 41,
114
+ "fps": 16
115
+ }
116
+ }
117
+ ]
118
+ },
119
+ "meta": {
120
+ "name": "[name]",
121
+ "version": "1.0"
122
+ }
123
+ }
LR_LORA_WAN_512/LR_LORA_WAN_512.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dbc88b12df43ca9730320d2d15a193fb8c5e1af9f0578ece33473aed1678ac29
3
+ size 4907438240
LR_LORA_WAN_512/LR_LORA_WAN_512_000001000.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5a464dcc7ffe73eef170ad7bb553e91ef679fecd7c9977fbe21f1732ff674bec
3
+ size 4907438240
LR_LORA_WAN_512/LR_LORA_WAN_512_000001500.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:429e91aaa1e29b5ed4f3cacc70cc8c901882d567c9cd9b350306bd516b354eda
3
+ size 4907438240
LR_LORA_WAN_512/LR_LORA_WAN_512_000002000.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:74a8e60c99af3974361dbfa28597366905c332a708daf1a7a693ae53ee3a6ebe
3
+ size 4907438240
LR_LORA_WAN_512/LR_LORA_WAN_512_000002500.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61e8ff37b47af64ab958dc9c2b0d816cadad6db2c536d4256313395e72e7df7b
3
+ size 4907438240
LR_LORA_WAN_512/config.yaml ADDED
@@ -0,0 +1,103 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ job: extension
2
+ config:
3
+ name: LR_LORA_WAN_512
4
+ process:
5
+ - type: diffusion_trainer
6
+ training_folder: /workspace/ai-toolkit/output
7
+ sqlite_db_path: /workspace/ai-toolkit/aitk_db.db
8
+ device: cuda
9
+ trigger_word: vmvm
10
+ performance_log_every: 10
11
+ network:
12
+ type: lora
13
+ linear: 512
14
+ linear_alpha: 512
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: 4
25
+ save_format: diffusers
26
+ push_to_hub: false
27
+ datasets:
28
+ - folder_path: /workspace/ai-toolkit/datasets/wexrct
29
+ mask_path: null
30
+ mask_min_value: 0.1
31
+ default_caption: vmvm
32
+ caption_ext: txt
33
+ caption_dropout_rate: 0.05
34
+ cache_latents_to_disk: false
35
+ is_reg: false
36
+ network_weight: 1
37
+ resolution:
38
+ - 1024
39
+ controls: []
40
+ shrink_video_to_frames: true
41
+ num_frames: 1
42
+ flip_x: false
43
+ flip_y: false
44
+ num_repeats: 2
45
+ train:
46
+ batch_size: 1
47
+ bypass_guidance_embedding: false
48
+ steps: 3000
49
+ gradient_accumulation: 1
50
+ train_unet: true
51
+ train_text_encoder: false
52
+ gradient_checkpointing: true
53
+ noise_scheduler: flowmatch
54
+ optimizer: adamw8bit
55
+ timestep_type: sigmoid
56
+ content_or_style: balanced
57
+ optimizer_params:
58
+ weight_decay: 0.0001
59
+ unload_text_encoder: false
60
+ cache_text_embeddings: false
61
+ lr: 0.0003
62
+ ema_config:
63
+ use_ema: false
64
+ ema_decay: 0.99
65
+ skip_first_sample: false
66
+ force_first_sample: false
67
+ disable_sampling: false
68
+ dtype: bf16
69
+ diff_output_preservation: false
70
+ diff_output_preservation_multiplier: 1
71
+ diff_output_preservation_class: person
72
+ switch_boundary_every: 1
73
+ loss_type: mse
74
+ do_differential_guidance: true
75
+ differential_guidance_scale: 3
76
+ logging:
77
+ log_every: 1
78
+ use_ui_logger: true
79
+ model:
80
+ name_or_path: Wan-AI/Wan2.1-T2V-14B-Diffusers
81
+ quantize: true
82
+ qtype: qfloat8
83
+ quantize_te: true
84
+ qtype_te: qfloat8
85
+ arch: wan21:14b
86
+ low_vram: false
87
+ model_kwargs: {}
88
+ sample:
89
+ sampler: flowmatch
90
+ sample_every: 250
91
+ width: 1024
92
+ height: 1024
93
+ samples: []
94
+ neg: ''
95
+ seed: 42
96
+ walk_seed: true
97
+ guidance_scale: 4
98
+ sample_steps: 25
99
+ num_frames: 41
100
+ fps: 16
101
+ meta:
102
+ name: LR_LORA_WAN_512
103
+ version: '1.0'
LR_LORA_WAN_512/log.txt ADDED
The diff for this file is too large to render. See raw diff
 
LR_LORA_WAN_512/loss_log.db ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b2591e60e6987fa047ffd5cd557bbc5d398ec2cf6c4eab7c321e3d64e3693a34
3
+ size 565248
LR_LORA_WAN_512/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7cf51ba84db8b722b38c7e048e8839aa114f35b3c134df4ae9e85faa0cbefec4
3
+ size 4984962189
LR_LORA_WAN_512/pid.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ 4692