arhiteector commited on
Commit
1f2e75c
·
verified ·
1 Parent(s): 44cb949

upload billy checkpoints (ignore sqlite wal/shm)

Browse files
.gitattributes CHANGED
@@ -40,3 +40,4 @@ LR_LORA_WAN_512/loss_log.db filter=lfs diff=lfs merge=lfs -text
40
  pretty512/loss_log.db filter=lfs diff=lfs merge=lfs -text
41
  mastur_copy/loss_log.db filter=lfs diff=lfs merge=lfs -text
42
  kurpa/loss_log.db filter=lfs diff=lfs merge=lfs -text
 
 
40
  pretty512/loss_log.db filter=lfs diff=lfs merge=lfs -text
41
  mastur_copy/loss_log.db filter=lfs diff=lfs merge=lfs -text
42
  kurpa/loss_log.db filter=lfs diff=lfs merge=lfs -text
43
+ billy/loss_log.db filter=lfs diff=lfs merge=lfs -text
billy/.job_config.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "job": "extension",
3
+ "config": {
4
+ "name": "billy",
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": 1000,
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/qwe",
35
+ "mask_path": null,
36
+ "mask_min_value": 0.1,
37
+ "default_caption": "vmvm woman",
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": 1
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": "weighted",
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": "Tongyi-MAI/Z-Image-Turbo",
94
+ "quantize": true,
95
+ "qtype": "qfloat8",
96
+ "quantize_te": true,
97
+ "qtype_te": "qfloat8",
98
+ "arch": "zimage:turbo",
99
+ "low_vram": false,
100
+ "model_kwargs": {},
101
+ "layer_offloading": false,
102
+ "layer_offloading_text_encoder_percent": 1,
103
+ "layer_offloading_transformer_percent": 1,
104
+ "assistant_lora_path": "ostris/zimage_turbo_training_adapter/zimage_turbo_training_adapter_v2.safetensors"
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": 1,
116
+ "sample_steps": 8,
117
+ "num_frames": 1,
118
+ "fps": 1
119
+ }
120
+ }
121
+ ]
122
+ },
123
+ "meta": {
124
+ "name": "[name]",
125
+ "version": "1.0"
126
+ }
127
+ }
billy/billy.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9057044fd1611cfb077e3128c50e2ff349d4c4aeb22a7ff54e52a3318a14f41f
3
+ size 2721118736
billy/billy_000001000.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4af890b3b74f5a7f245c8f5cf1c0c5f51e81803258463dee66afb4922abe8839
3
+ size 2721118736
billy/billy_000002000.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:635c7ec88d90bf4444801a6fb780acf9104eab33325833b1693fbd6afa959308
3
+ size 2721118736
billy/config.yaml ADDED
@@ -0,0 +1,107 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ job: extension
2
+ config:
3
+ name: billy
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: 1000
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/qwe
29
+ mask_path: null
30
+ mask_min_value: 0.1
31
+ default_caption: vmvm woman
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: 1
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: weighted
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: Tongyi-MAI/Z-Image-Turbo
81
+ quantize: true
82
+ qtype: qfloat8
83
+ quantize_te: true
84
+ qtype_te: qfloat8
85
+ arch: zimage:turbo
86
+ low_vram: false
87
+ model_kwargs: {}
88
+ layer_offloading: false
89
+ layer_offloading_text_encoder_percent: 1
90
+ layer_offloading_transformer_percent: 1
91
+ assistant_lora_path: ostris/zimage_turbo_training_adapter/zimage_turbo_training_adapter_v2.safetensors
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: 1
102
+ sample_steps: 8
103
+ num_frames: 1
104
+ fps: 1
105
+ meta:
106
+ name: billy
107
+ version: '1.0'
billy/log.txt ADDED
The diff for this file is too large to render. See raw diff
 
billy/loss_log.db ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:19327684df264f0fd83b761899b8d2806244342abd80334cff172556b952b07c
3
+ size 565248
billy/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50c64cdbc78e8e5e022110fbfeb99c1c46c584b10b233406d77c40a02be88260
3
+ size 2764142821
billy/pid.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ 1670