Add files using upload-large-folder tool
Browse files- models/7b/checklist.chk +3 -0
- models/7b/consolidate_params.json +8 -0
- models/7b/consolidated.pth +3 -0
- models/7b/params.json +169 -0
- tokenizer/checklist.chk +3 -0
- tokenizer/text_tokenizer.json +0 -0
- tokenizer/vqgan.ckpt +3 -0
- tokenizer/vqgan.yaml +57 -0
models/7b/checklist.chk
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
60333d9acd866e4b5e6690ecabee3b65 consolidated.pth
|
2 |
+
b2cbf6940c157b6e969f6263388efcc3 consolidate_params.json
|
3 |
+
8bc0e859d4afa00f5f40a9de296eef8e params.json
|
models/7b/consolidate_params.json
ADDED
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"dtype": "bf16",
|
3 |
+
"model_parallel_size": 1,
|
4 |
+
"on_gpu": true,
|
5 |
+
"src": "/fsx-onellm/rpasunuru/SFT/v2.1_textpp_7b_1366k_sftv1.4_exp1/v2.1_textpp_7b_1366k_sftv1.4_exp1_run000/checkpoints/checkpoint_0001200_noimggen/",
|
6 |
+
"tgt": "/fsx-onellm/rpasunuru/SFT/v2.1_textpp_7b_1366k_sftv1.4_exp1/v2.1_textpp_7b_1366k_sftv1.4_exp1_run000/checkpoints/checkpoint_0001200_noimggen_consolidated/",
|
7 |
+
"tokenizer_path": null
|
8 |
+
}
|
models/7b/consolidated.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:dc5978837b6c879570b17e64caa0e6c2a9458a4c5fb0255dc0dd526492e58bcb
|
3 |
+
size 14026453651
|
models/7b/params.json
ADDED
@@ -0,0 +1,169 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"async_checkpointing": false,
|
3 |
+
"async_eval_ngpus": -1,
|
4 |
+
"batch_size": 2,
|
5 |
+
"data": "",
|
6 |
+
"disable_logging": false,
|
7 |
+
"disable_workers_print": false,
|
8 |
+
"dtype": "bf16",
|
9 |
+
"dump_after_steps": 0,
|
10 |
+
"dump_dir": "/fsx-onellm/rpasunuru/SFT/v2.1_textpp_7b_1366k_sftv1.4_exp1/v2.1_textpp_7b_1366k_sftv1.4_exp1_run000",
|
11 |
+
"dump_freq": 400,
|
12 |
+
"dump_profile_traces": false,
|
13 |
+
"enable_loss_tracker": false,
|
14 |
+
"epochs": -1,
|
15 |
+
"eval_freq": 400,
|
16 |
+
"exp_id": "",
|
17 |
+
"exp_name": "",
|
18 |
+
"finetuning_dir": "/fsx-onellm/shared/from_rsc/v2.1_7b_dr_qk_zloss_linear_zero3_sft_optiml_textpp_run000_checkpoint_1366000",
|
19 |
+
"fp32_reduce_scatter": "all",
|
20 |
+
"gpu_check_level": 3,
|
21 |
+
"image_loss_weight": 1.0,
|
22 |
+
"image_text_rotation_prob": 0.0,
|
23 |
+
"instruct": {
|
24 |
+
"no_loss_prompt": true,
|
25 |
+
"no_loss_truncated": false,
|
26 |
+
"use_eot": true
|
27 |
+
},
|
28 |
+
"instruct_data": "/fsx-onellm/shared/finetuning/sft_v1.4_data/splits/train/long_caption:2.92,/fsx-onellm/shared/finetuning/sft_v1.4_data/splits/train/vqa:4.59,/fsx-onellm/shared/finetuning/sft_v1.4_data/splits/train/text2image:10.44,/fsx-onellm/shared/finetuning/sft_v1.4_data/splits/train/llama2_rjv6_helpful:43.27,/fsx-onellm/shared/finetuning/sft_v1.4_data/splits/train/code_llama:0.51,/fsx-onellm/shared/finetuning/sft_v1.4_data/splits/train/interleaved_batch1-17:27.45,/fsx-onellm/shared/finetuning/sft_v1.4_data/splits/train/image_dialogue:7.46,/fsx-onellm/shared/finetuning/sft_v1.4_data/splits/train/llama2_rjv6_harmless:0.97,/fsx-onellm/shared/finetuning/sft_v1.4_data/splits/train/cybersec_safety:0.33,/fsx-onellm/shared/finetuning/sft_v1.4_data/splits/train/onellm_multimodal_safety:0.86,/fsx-onellm/shared/finetuning/sft_v1.4_data/splits/train/autosafety:0.51,/fsx-onellm/shared/finetuning/sft_v1.4_data/splits/train/rainbow_safety:0.10,/fsx-onellm/shared/finetuning/sft_v1.4_data/splits/train/genai_safety:0.58",
|
29 |
+
"iter_gopher": {
|
30 |
+
"buffer_size": 16,
|
31 |
+
"max_precompute": 10,
|
32 |
+
"n_chars_by_tok": 15,
|
33 |
+
"n_seqs_to_concat": 10,
|
34 |
+
"num_processes": 1
|
35 |
+
},
|
36 |
+
"iter_jsonl": {
|
37 |
+
"buffer_size": 64,
|
38 |
+
"same_data": false
|
39 |
+
},
|
40 |
+
"iter_multi": {
|
41 |
+
"buffer_size": 512,
|
42 |
+
"ignore_extra_chunks": true,
|
43 |
+
"max_precompute": 20,
|
44 |
+
"multiprocess": true
|
45 |
+
},
|
46 |
+
"iter_type": "multi",
|
47 |
+
"keep_checkpoints_every_steps": 400,
|
48 |
+
"keep_eval_checkpoints": true,
|
49 |
+
"keep_n_last_checkpoints": 2,
|
50 |
+
"log_all_steps": false,
|
51 |
+
"log_freq": 10,
|
52 |
+
"log_updates": true,
|
53 |
+
"log_wandb": false,
|
54 |
+
"loss_rescaling": false,
|
55 |
+
"model": {
|
56 |
+
"add_extra_toks": "0",
|
57 |
+
"alpha_depth": "disabled",
|
58 |
+
"attn_dropout": 0,
|
59 |
+
"attn_to_keep": "all",
|
60 |
+
"custom_bwd": false,
|
61 |
+
"dim": 4096,
|
62 |
+
"dropout": 0.05,
|
63 |
+
"efficient_attn": "flash",
|
64 |
+
"emb_dropout": 0,
|
65 |
+
"ffn_dim_multiplier": 1.0,
|
66 |
+
"ffn_dropout": 0,
|
67 |
+
"full_logging_n_layers": 4,
|
68 |
+
"fuse_sequence_parallel": false,
|
69 |
+
"init": {
|
70 |
+
"coeff_std": null,
|
71 |
+
"depth_last": false,
|
72 |
+
"fixed_std": null,
|
73 |
+
"no_init": false,
|
74 |
+
"pos_init_scalar": null,
|
75 |
+
"use_depth": "current",
|
76 |
+
"use_gaussian": true
|
77 |
+
},
|
78 |
+
"layer_ckpt": "none",
|
79 |
+
"linear_residual_dropout": false,
|
80 |
+
"loss_parallel": false,
|
81 |
+
"max_length": 2048,
|
82 |
+
"multiple_of": 256,
|
83 |
+
"n_heads": 32,
|
84 |
+
"n_kv_heads": null,
|
85 |
+
"n_layers": 32,
|
86 |
+
"non_linearity": "swiglu",
|
87 |
+
"norm_affine": true,
|
88 |
+
"norm_eps": 1e-05,
|
89 |
+
"norm_type": "rmsnorm",
|
90 |
+
"output_dropout": 0,
|
91 |
+
"output_size": -1,
|
92 |
+
"pre_norm": true,
|
93 |
+
"qk_normalization": true,
|
94 |
+
"recompute_attn": true,
|
95 |
+
"recompute_fc1_out": true,
|
96 |
+
"recompute_fc3_out": true,
|
97 |
+
"residual_dropout": 0.0,
|
98 |
+
"rope_theta": 10000.0,
|
99 |
+
"sequence_parallel": false,
|
100 |
+
"swin_norm": false,
|
101 |
+
"turn_eos_token": "<eos>",
|
102 |
+
"use_rope": true,
|
103 |
+
"vocab_size": 65536
|
104 |
+
},
|
105 |
+
"model_parallel_size": 1,
|
106 |
+
"no_final_ckpt": false,
|
107 |
+
"num_retrieved_docs": 0,
|
108 |
+
"old_mp": -1,
|
109 |
+
"old_world_size": -1,
|
110 |
+
"optim": {
|
111 |
+
"beta1": 0.9,
|
112 |
+
"beta2": 0.95,
|
113 |
+
"clip": 1.0,
|
114 |
+
"cosine_theta": 1.0,
|
115 |
+
"cycle_length": 1.0,
|
116 |
+
"epsilon": 1e-08,
|
117 |
+
"exp_factor": 0.5,
|
118 |
+
"lr": 1e-05,
|
119 |
+
"lr_min_ratio": 0.1,
|
120 |
+
"scheduler": "cosine",
|
121 |
+
"use_deprecated_optim": false,
|
122 |
+
"warmup": 100,
|
123 |
+
"weight_decay": 0.1
|
124 |
+
},
|
125 |
+
"periodic_gpu_check": true,
|
126 |
+
"profile_freq": -1,
|
127 |
+
"reshard_after_forward": false,
|
128 |
+
"restore_dataloader_position": false,
|
129 |
+
"retrieval_prob": 0.0,
|
130 |
+
"rlhf": null,
|
131 |
+
"root_dump_dir": "",
|
132 |
+
"save_optimizer_states": true,
|
133 |
+
"seq_len": 4096,
|
134 |
+
"slurm": {
|
135 |
+
"global_rank": 0,
|
136 |
+
"is_slurm_job": true,
|
137 |
+
"world_size": 64
|
138 |
+
},
|
139 |
+
"steps": 1200,
|
140 |
+
"tokenizer": "/fsx-onellm/rpasunuru/models/cm3z/cm3v2_7b_placeholder/gpt2-unified-image-sentinel.json",
|
141 |
+
"tokenizer_dir": "/fsx/guismay/data/large_experiments/fair_llm/datasets/tokenizers",
|
142 |
+
"torch_seed": -1,
|
143 |
+
"unlimited_steps": false,
|
144 |
+
"use_hf_tokenizer": true,
|
145 |
+
"valid": {
|
146 |
+
"batch_size": 32,
|
147 |
+
"debug": false,
|
148 |
+
"majority_voting": 0,
|
149 |
+
"n_batches": 100,
|
150 |
+
"onellm_eval": false,
|
151 |
+
"onellm_eval_media_storage": "",
|
152 |
+
"ppl_files_str": "",
|
153 |
+
"prompt_path": "",
|
154 |
+
"prompt_templates": "{}",
|
155 |
+
"random_fewshots": false,
|
156 |
+
"seq_len": 2048,
|
157 |
+
"tasks_root_dir": "",
|
158 |
+
"tasks_str": "",
|
159 |
+
"temperature": 1.0,
|
160 |
+
"top_k": 0,
|
161 |
+
"top_p": 0.0,
|
162 |
+
"use_sampling": false,
|
163 |
+
"write_eval": false
|
164 |
+
},
|
165 |
+
"wandb_entity": "violet-zct",
|
166 |
+
"wandb_project": "instruct_sft",
|
167 |
+
"water_marking_codes_str": null,
|
168 |
+
"z_loss_weight": 0.0001
|
169 |
+
}
|
tokenizer/checklist.chk
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
170a932b687671a4e676f3bf69147295 text_tokenizer.json
|
2 |
+
1a559fb5dab4d351d19496ae89da1db1 vqgan.ckpt
|
3 |
+
25724c8110d6adabc9130a123b4b922e vqgan.yaml
|
tokenizer/text_tokenizer.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
tokenizer/vqgan.ckpt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4ede986bf6b171db3081ce171ad88e4ac970793cea14c180b3e5ac5105f4cb43
|
3 |
+
size 281270377
|
tokenizer/vqgan.yaml
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
model:
|
2 |
+
base_learning_rate: 4.5e-06
|
3 |
+
target: taming.models.vqgan.VQModel
|
4 |
+
params:
|
5 |
+
embed_dim: 256
|
6 |
+
n_embed: 8192
|
7 |
+
ddconfig:
|
8 |
+
double_z: false
|
9 |
+
z_channels: 256
|
10 |
+
resolution: 512
|
11 |
+
in_channels: 3
|
12 |
+
out_ch: 3
|
13 |
+
ch: 128
|
14 |
+
ch_mult:
|
15 |
+
- 1
|
16 |
+
- 1
|
17 |
+
- 2
|
18 |
+
- 2
|
19 |
+
- 4
|
20 |
+
num_res_blocks: 2
|
21 |
+
attn_resolutions: []
|
22 |
+
dropout: 0.0
|
23 |
+
lossconfig:
|
24 |
+
target: taming.modules.losses.vqperceptual_vit_vqgan.VQLPIPSWithDiscriminator
|
25 |
+
params:
|
26 |
+
disc_start: 100001
|
27 |
+
perceptual_weight: 1.0
|
28 |
+
adversarial_weight: 0.5
|
29 |
+
disc_params:
|
30 |
+
size: 512
|
31 |
+
ckpt_path: manifold://fair_onellm_checkpoints/tree/v2/tokenizer/vqgan_wm_0209.ckpt
|
32 |
+
data:
|
33 |
+
target: main.DataModuleFromConfig
|
34 |
+
params:
|
35 |
+
batch_size: 4
|
36 |
+
num_workers: 10
|
37 |
+
image_size: 512
|
38 |
+
filter_image_size: 512
|
39 |
+
dataset: coco
|
40 |
+
aesthetics_th: 0
|
41 |
+
clipsim_th: 0
|
42 |
+
--distributed-world-size: null
|
43 |
+
'32': null
|
44 |
+
--distributed-port: null
|
45 |
+
'17338': null
|
46 |
+
--save-dir: null
|
47 |
+
/checkpoint/shellysheynin/shutterstock/512x512_1024tokens_4node_shutterstock_laion_no_attn_styleGAN:
|
48 |
+
log_every-500:
|
49 |
+
ngpu32: null
|
50 |
+
--tensorboard-logdir: null
|
51 |
+
/checkpoint/shellysheynin/tensorboard_logs/2023-03-30/512x512_1024tokens_4node_shutterstock_laion_no_attn_styleGAN:
|
52 |
+
log_every-500:
|
53 |
+
ngpu32: null
|
54 |
+
'14561': null
|
55 |
+
/checkpoint/shellysheynin/tensorboard_logs/2023-04-02/512x512_1024tokens_4node_shutterstock_laion_no_attn_styleGAN:
|
56 |
+
log_every-500:
|
57 |
+
ngpu32: null
|