Hugging Face's logo

lodestones
/
chroma-debug-development-only

Model card Files Community
lodestones commited on
Commit
81310e0
·
verified ·
1 Parent(s): a751e0c

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. priming_e6/2025-02-09_03-51-08.pth +3 -0
  2. priming_e6/training_config.json +117 -0
priming_e6/2025-02-09_03-51-08.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bdf77064e64ba5a93362574fa10d46823a132de2923604da279650a7934a83db
3
+ size 17800230264
priming_e6/training_config.json ADDED
@@ -0,0 +1,117 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "training": {
3
+ "master_seed": 0,
4
+ "cache_minibatch": 16,
5
+ "train_minibatch": 16,
6
+ "offload_param_count": 17000000000,
7
+ "lr": 1e-05,
8
+ "weight_decay": 0.0,
9
+ "warmup_steps": 5,
10
+ "change_layer_every": 100,
11
+ "trained_single_blocks": 38,
12
+ "trained_double_blocks": 19,
13
+ "save_every": 50,
14
+ "save_folder": "/workspace/checkpoint/training",
15
+ "wandb_key": "08f134109a388d4d784b091c1c145bf0a32c5de4",
16
+ "wandb_project": "optimal transport unlocked",
17
+ "wandb_run": "priming",
18
+ "wandb_entity": null,
19
+ "hf_repo_id": null,
20
+ "hf_token": null
21
+ },
22
+ "inference": {
23
+ "inference_every": 20,
24
+ "inference_folder": "/workspace/checkpoint/preview",
25
+ "steps": 20,
26
+ "guidance": 0,
27
+ "cfg": 3,
28
+ "prompts": [
29
+ "a cute cat sat on a mat while receiving a head pat from his owner called Matt",
30
+ "baked potato, on the space floating orbiting around the earth"
31
+ ],
32
+ "first_n_steps_wo_cfg": 0,
33
+ "image_dim": [
34
+ 1024,
35
+ 1024
36
+ ],
37
+ "t5_max_length": 512
38
+ },
39
+ "extra_inference_config": [
40
+ {
41
+ "inference_every": 20,
42
+ "inference_folder": "/workspace/checkpoint/preview",
43
+ "steps": 20,
44
+ "guidance": 0,
45
+ "cfg": 4,
46
+ "prompts": [
47
+ "a cute cat sat on a mat while receiving a head pat from his owner called Matt",
48
+ "baked potato, on the space floating orbiting around the earth"
49
+ ],
50
+ "first_n_steps_wo_cfg": 0,
51
+ "image_dim": [
52
+ 1024,
53
+ 1024
54
+ ],
55
+ "t5_max_length": 512
56
+ },
57
+ {
58
+ "inference_every": 20,
59
+ "inference_folder": "/workspace/checkpoint/preview",
60
+ "steps": 20,
61
+ "guidance": 0,
62
+ "cfg": 5,
63
+ "prompts": [
64
+ "a cute cat sat on a mat while receiving a head pat from his owner called Matt",
65
+ "baked potato, on the space floating orbiting around the earth"
66
+ ],
67
+ "first_n_steps_wo_cfg": 0,
68
+ "image_dim": [
69
+ 1024,
70
+ 1024
71
+ ],
72
+ "t5_max_length": 512
73
+ },
74
+ {
75
+ "inference_every": 20,
76
+ "inference_folder": "/workspace/checkpoint/preview",
77
+ "steps": 20,
78
+ "guidance": 0,
79
+ "cfg": 6,
80
+ "prompts": [
81
+ "a cute cat sat on a mat while receiving a head pat from his owner called Matt",
82
+ "baked potato, on the space floating orbiting around the earth"
83
+ ],
84
+ "first_n_steps_wo_cfg": 0,
85
+ "image_dim": [
86
+ 1024,
87
+ 1024
88
+ ],
89
+ "t5_max_length": 512
90
+ }
91
+ ],
92
+ "dataloader": {
93
+ "batch_size": 1024,
94
+ "jsonl_metadata_path": "/workspace/all_rounder.jsonl",
95
+ "image_folder_path": "",
96
+ "base_resolution": [
97
+ 512
98
+ ],
99
+ "shuffle_tags": true,
100
+ "tag_drop_percentage": 0.0,
101
+ "uncond_percentage": 0.1,
102
+ "resolution_step": 64,
103
+ "num_workers": 1,
104
+ "prefetch_factor": 1,
105
+ "ratio_cutoff": 2.0,
106
+ "thread_per_worker": 2
107
+ },
108
+ "model": {
109
+ "chroma_path": "/workspace/checkpoint/chroma-unlocked-v5.safetensors",
110
+ "vae_path": "/workspace/FLUX.1-schnell/ae.safetensors",
111
+ "t5_path": "/workspace/FLUX.1-schnell/text_encoder_2",
112
+ "t5_config_path": "/workspace/FLUX.1-schnell/text_encoder_2/config.json",
113
+ "t5_tokenizer_path": "/workspace/FLUX.1-schnell/tokenizer_2",
114
+ "t5_to_8bit": true,
115
+ "t5_max_length": 512
116
+ }
117
+ }