1d6ee5839cd29b66481f3a49c92e5fd9cad77de5c649f2d51506f3d16a99fe72
Browse files- README.md +3 -3
- config.json +54 -55
- generation_config.json +1 -1
- merges.txt +0 -0
- smash_config.json +5 -2
- special_tokens_map.json +34 -0
- tokenizer.json +0 -0
- tokenizer_config.json +154 -0
- vocab.json +0 -0
README.md
CHANGED
@@ -14,8 +14,8 @@ tags:
|
|
14 |
<!-- header start -->
|
15 |
<!-- 200823 -->
|
16 |
<div style="width: auto; margin-left: auto; margin-right: auto">
|
17 |
-
<a href="https://
|
18 |
-
<img src="https://imgur.com/
|
19 |
</a>
|
20 |
</div>
|
21 |
<!-- header end -->
|
@@ -82,4 +82,4 @@ The license of the smashed model follows the license of the original model. Plea
|
|
82 |
## Want to compress other models?
|
83 |
|
84 |
- Contact us and tell us which model to compress next [here](https://www.pruna.ai/contact).
|
85 |
-
-
|
|
|
14 |
<!-- header start -->
|
15 |
<!-- 200823 -->
|
16 |
<div style="width: auto; margin-left: auto; margin-right: auto">
|
17 |
+
<a href="https://www.pruna.ai/" target="_blank" rel="noopener noreferrer">
|
18 |
+
<img src="https://i.imgur.com/eDAlcgk.png" alt="PrunaAI" style="width: 100%; min-width: 400px; display: block; margin: auto;">
|
19 |
</a>
|
20 |
</div>
|
21 |
<!-- header end -->
|
|
|
82 |
## Want to compress other models?
|
83 |
|
84 |
- Contact us and tell us which model to compress next [here](https://www.pruna.ai/contact).
|
85 |
+
- Request access to easily compress your own AI models [here](https://z0halsaff74.typeform.com/pruna-access?typeform-source=www.pruna.ai).
|
config.json
CHANGED
@@ -1,57 +1,56 @@
|
|
1 |
{
|
2 |
-
|
3 |
-
|
4 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
5 |
],
|
6 |
-
"
|
7 |
-
"
|
8 |
-
"
|
9 |
-
"
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
-
|
14 |
-
|
15 |
-
|
16 |
-
|
17 |
-
|
18 |
-
"
|
19 |
-
|
20 |
-
|
21 |
-
|
22 |
-
|
23 |
-
|
24 |
-
|
25 |
-
|
26 |
-
|
27 |
-
"bnb_4bit_compute_dtype": "bfloat16",
|
28 |
-
"bnb_4bit_quant_storage": "uint8",
|
29 |
-
"bnb_4bit_quant_type": "fp4",
|
30 |
-
"bnb_4bit_use_double_quant": false,
|
31 |
-
"llm_int8_enable_fp32_cpu_offload": false,
|
32 |
-
"llm_int8_has_fp16_weight": false,
|
33 |
-
"llm_int8_skip_modules": [
|
34 |
-
"lm_head"
|
35 |
-
],
|
36 |
-
"llm_int8_threshold": 6.0,
|
37 |
-
"load_in_4bit": false,
|
38 |
-
"load_in_8bit": true,
|
39 |
-
"quant_method": "bitsandbytes"
|
40 |
-
},
|
41 |
-
"rms_norm_eps": 1e-05,
|
42 |
-
"rope_interleaved": false,
|
43 |
-
"rope_scaling": null,
|
44 |
-
"rope_theta": 100000,
|
45 |
-
"tie_word_embeddings": true,
|
46 |
-
"torch_dtype": "float16",
|
47 |
-
"transformers.js_config": {
|
48 |
-
"kv_cache_dtype": {
|
49 |
-
"fp16": "float16",
|
50 |
-
"q4f16": "float16"
|
51 |
-
}
|
52 |
-
},
|
53 |
-
"transformers_version": "4.46.2",
|
54 |
-
"use_cache": true,
|
55 |
-
"vocab_size": 49152,
|
56 |
-
"api_key": null
|
57 |
-
}
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "/home/ubuntu/.cache/pruna/tmp6r_sc64vvr7zr28q",
|
3 |
+
"architectures": [
|
4 |
+
"LlamaForCausalLM"
|
5 |
+
],
|
6 |
+
"attention_bias": false,
|
7 |
+
"attention_dropout": 0.0,
|
8 |
+
"bos_token_id": 1,
|
9 |
+
"eos_token_id": 2,
|
10 |
+
"head_dim": 64,
|
11 |
+
"hidden_act": "silu",
|
12 |
+
"hidden_size": 576,
|
13 |
+
"initializer_range": 0.041666666666666664,
|
14 |
+
"intermediate_size": 1536,
|
15 |
+
"is_llama_config": true,
|
16 |
+
"max_position_embeddings": 8192,
|
17 |
+
"mlp_bias": false,
|
18 |
+
"model_type": "llama",
|
19 |
+
"num_attention_heads": 9,
|
20 |
+
"num_hidden_layers": 30,
|
21 |
+
"num_key_value_heads": 3,
|
22 |
+
"pad_token_id": 2,
|
23 |
+
"pretraining_tp": 1,
|
24 |
+
"quantization_config": {
|
25 |
+
"_load_in_4bit": false,
|
26 |
+
"_load_in_8bit": true,
|
27 |
+
"bnb_4bit_compute_dtype": "bfloat16",
|
28 |
+
"bnb_4bit_quant_storage": "uint8",
|
29 |
+
"bnb_4bit_quant_type": "fp4",
|
30 |
+
"bnb_4bit_use_double_quant": false,
|
31 |
+
"llm_int8_enable_fp32_cpu_offload": false,
|
32 |
+
"llm_int8_has_fp16_weight": false,
|
33 |
+
"llm_int8_skip_modules": [
|
34 |
+
"lm_head"
|
35 |
],
|
36 |
+
"llm_int8_threshold": 6.0,
|
37 |
+
"load_in_4bit": false,
|
38 |
+
"load_in_8bit": true,
|
39 |
+
"quant_method": "bitsandbytes"
|
40 |
+
},
|
41 |
+
"rms_norm_eps": 1e-05,
|
42 |
+
"rope_interleaved": false,
|
43 |
+
"rope_scaling": null,
|
44 |
+
"rope_theta": 100000,
|
45 |
+
"tie_word_embeddings": true,
|
46 |
+
"torch_dtype": "float16",
|
47 |
+
"transformers.js_config": {
|
48 |
+
"kv_cache_dtype": {
|
49 |
+
"fp16": "float16",
|
50 |
+
"q4f16": "float16"
|
51 |
+
}
|
52 |
+
},
|
53 |
+
"transformers_version": "4.46.3",
|
54 |
+
"use_cache": true,
|
55 |
+
"vocab_size": 49152
|
56 |
+
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
generation_config.json
CHANGED
@@ -3,5 +3,5 @@
|
|
3 |
"bos_token_id": 1,
|
4 |
"eos_token_id": 2,
|
5 |
"pad_token_id": 2,
|
6 |
-
"transformers_version": "4.46.
|
7 |
}
|
|
|
3 |
"bos_token_id": 1,
|
4 |
"eos_token_id": 2,
|
5 |
"pad_token_id": 2,
|
6 |
+
"transformers_version": "4.46.3"
|
7 |
}
|
merges.txt
ADDED
The diff for this file is too large to render.
See raw diff
|
|
smash_config.json
CHANGED
@@ -3,13 +3,16 @@
|
|
3 |
"comp_ctranslate_active": false,
|
4 |
"comp_cwhisper_active": false,
|
5 |
"comp_diffusers2_active": false,
|
|
|
6 |
"comp_ifw_active": false,
|
|
|
7 |
"comp_onediff_active": false,
|
8 |
"comp_step_caching_active": false,
|
9 |
"comp_torch_compile_active": false,
|
10 |
"comp_ws2t_active": false,
|
11 |
"comp_x-fast_active": false,
|
12 |
"prune_torch-structured_active": false,
|
|
|
13 |
"quant_aqlm_active": false,
|
14 |
"quant_awq_active": false,
|
15 |
"quant_gptq_active": false,
|
@@ -28,8 +31,8 @@
|
|
28 |
"quant_llm-int8_weight_bits": 8,
|
29 |
"max_batch_size": 1,
|
30 |
"device": "cuda",
|
31 |
-
"cache_dir": "/
|
32 |
"task": "",
|
33 |
-
"save_load_fn": "
|
34 |
"save_load_fn_args": {}
|
35 |
}
|
|
|
3 |
"comp_ctranslate_active": false,
|
4 |
"comp_cwhisper_active": false,
|
5 |
"comp_diffusers2_active": false,
|
6 |
+
"comp_flux_caching_active": false,
|
7 |
"comp_ifw_active": false,
|
8 |
+
"comp_ipex_llm_active": false,
|
9 |
"comp_onediff_active": false,
|
10 |
"comp_step_caching_active": false,
|
11 |
"comp_torch_compile_active": false,
|
12 |
"comp_ws2t_active": false,
|
13 |
"comp_x-fast_active": false,
|
14 |
"prune_torch-structured_active": false,
|
15 |
+
"prune_torch-unstructured_active": false,
|
16 |
"quant_aqlm_active": false,
|
17 |
"quant_awq_active": false,
|
18 |
"quant_gptq_active": false,
|
|
|
31 |
"quant_llm-int8_weight_bits": 8,
|
32 |
"max_batch_size": 1,
|
33 |
"device": "cuda",
|
34 |
+
"cache_dir": "/home/ubuntu/.cache/pruna/tmp6r_sc64v",
|
35 |
"task": "",
|
36 |
+
"save_load_fn": "llm-int8",
|
37 |
"save_load_fn_args": {}
|
38 |
}
|
special_tokens_map.json
ADDED
@@ -0,0 +1,34 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"additional_special_tokens": [
|
3 |
+
"<|im_start|>",
|
4 |
+
"<|im_end|>"
|
5 |
+
],
|
6 |
+
"bos_token": {
|
7 |
+
"content": "<|im_start|>",
|
8 |
+
"lstrip": false,
|
9 |
+
"normalized": false,
|
10 |
+
"rstrip": false,
|
11 |
+
"single_word": false
|
12 |
+
},
|
13 |
+
"eos_token": {
|
14 |
+
"content": "<|im_end|>",
|
15 |
+
"lstrip": false,
|
16 |
+
"normalized": false,
|
17 |
+
"rstrip": false,
|
18 |
+
"single_word": false
|
19 |
+
},
|
20 |
+
"pad_token": {
|
21 |
+
"content": "<|im_end|>",
|
22 |
+
"lstrip": false,
|
23 |
+
"normalized": false,
|
24 |
+
"rstrip": false,
|
25 |
+
"single_word": false
|
26 |
+
},
|
27 |
+
"unk_token": {
|
28 |
+
"content": "<|endoftext|>",
|
29 |
+
"lstrip": false,
|
30 |
+
"normalized": false,
|
31 |
+
"rstrip": false,
|
32 |
+
"single_word": false
|
33 |
+
}
|
34 |
+
}
|
tokenizer.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
tokenizer_config.json
ADDED
@@ -0,0 +1,154 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"add_prefix_space": false,
|
3 |
+
"added_tokens_decoder": {
|
4 |
+
"0": {
|
5 |
+
"content": "<|endoftext|>",
|
6 |
+
"lstrip": false,
|
7 |
+
"normalized": false,
|
8 |
+
"rstrip": false,
|
9 |
+
"single_word": false,
|
10 |
+
"special": true
|
11 |
+
},
|
12 |
+
"1": {
|
13 |
+
"content": "<|im_start|>",
|
14 |
+
"lstrip": false,
|
15 |
+
"normalized": false,
|
16 |
+
"rstrip": false,
|
17 |
+
"single_word": false,
|
18 |
+
"special": true
|
19 |
+
},
|
20 |
+
"2": {
|
21 |
+
"content": "<|im_end|>",
|
22 |
+
"lstrip": false,
|
23 |
+
"normalized": false,
|
24 |
+
"rstrip": false,
|
25 |
+
"single_word": false,
|
26 |
+
"special": true
|
27 |
+
},
|
28 |
+
"3": {
|
29 |
+
"content": "<repo_name>",
|
30 |
+
"lstrip": false,
|
31 |
+
"normalized": false,
|
32 |
+
"rstrip": false,
|
33 |
+
"single_word": false,
|
34 |
+
"special": true
|
35 |
+
},
|
36 |
+
"4": {
|
37 |
+
"content": "<reponame>",
|
38 |
+
"lstrip": false,
|
39 |
+
"normalized": false,
|
40 |
+
"rstrip": false,
|
41 |
+
"single_word": false,
|
42 |
+
"special": true
|
43 |
+
},
|
44 |
+
"5": {
|
45 |
+
"content": "<file_sep>",
|
46 |
+
"lstrip": false,
|
47 |
+
"normalized": false,
|
48 |
+
"rstrip": false,
|
49 |
+
"single_word": false,
|
50 |
+
"special": true
|
51 |
+
},
|
52 |
+
"6": {
|
53 |
+
"content": "<filename>",
|
54 |
+
"lstrip": false,
|
55 |
+
"normalized": false,
|
56 |
+
"rstrip": false,
|
57 |
+
"single_word": false,
|
58 |
+
"special": true
|
59 |
+
},
|
60 |
+
"7": {
|
61 |
+
"content": "<gh_stars>",
|
62 |
+
"lstrip": false,
|
63 |
+
"normalized": false,
|
64 |
+
"rstrip": false,
|
65 |
+
"single_word": false,
|
66 |
+
"special": true
|
67 |
+
},
|
68 |
+
"8": {
|
69 |
+
"content": "<issue_start>",
|
70 |
+
"lstrip": false,
|
71 |
+
"normalized": false,
|
72 |
+
"rstrip": false,
|
73 |
+
"single_word": false,
|
74 |
+
"special": true
|
75 |
+
},
|
76 |
+
"9": {
|
77 |
+
"content": "<issue_comment>",
|
78 |
+
"lstrip": false,
|
79 |
+
"normalized": false,
|
80 |
+
"rstrip": false,
|
81 |
+
"single_word": false,
|
82 |
+
"special": true
|
83 |
+
},
|
84 |
+
"10": {
|
85 |
+
"content": "<issue_closed>",
|
86 |
+
"lstrip": false,
|
87 |
+
"normalized": false,
|
88 |
+
"rstrip": false,
|
89 |
+
"single_word": false,
|
90 |
+
"special": true
|
91 |
+
},
|
92 |
+
"11": {
|
93 |
+
"content": "<jupyter_start>",
|
94 |
+
"lstrip": false,
|
95 |
+
"normalized": false,
|
96 |
+
"rstrip": false,
|
97 |
+
"single_word": false,
|
98 |
+
"special": true
|
99 |
+
},
|
100 |
+
"12": {
|
101 |
+
"content": "<jupyter_text>",
|
102 |
+
"lstrip": false,
|
103 |
+
"normalized": false,
|
104 |
+
"rstrip": false,
|
105 |
+
"single_word": false,
|
106 |
+
"special": true
|
107 |
+
},
|
108 |
+
"13": {
|
109 |
+
"content": "<jupyter_code>",
|
110 |
+
"lstrip": false,
|
111 |
+
"normalized": false,
|
112 |
+
"rstrip": false,
|
113 |
+
"single_word": false,
|
114 |
+
"special": true
|
115 |
+
},
|
116 |
+
"14": {
|
117 |
+
"content": "<jupyter_output>",
|
118 |
+
"lstrip": false,
|
119 |
+
"normalized": false,
|
120 |
+
"rstrip": false,
|
121 |
+
"single_word": false,
|
122 |
+
"special": true
|
123 |
+
},
|
124 |
+
"15": {
|
125 |
+
"content": "<jupyter_script>",
|
126 |
+
"lstrip": false,
|
127 |
+
"normalized": false,
|
128 |
+
"rstrip": false,
|
129 |
+
"single_word": false,
|
130 |
+
"special": true
|
131 |
+
},
|
132 |
+
"16": {
|
133 |
+
"content": "<empty_output>",
|
134 |
+
"lstrip": false,
|
135 |
+
"normalized": false,
|
136 |
+
"rstrip": false,
|
137 |
+
"single_word": false,
|
138 |
+
"special": true
|
139 |
+
}
|
140 |
+
},
|
141 |
+
"additional_special_tokens": [
|
142 |
+
"<|im_start|>",
|
143 |
+
"<|im_end|>"
|
144 |
+
],
|
145 |
+
"bos_token": "<|im_start|>",
|
146 |
+
"chat_template": "{% for message in messages %}{% if loop.first and messages[0]['role'] != 'system' %}{{ '<|im_start|>system\nYou are a helpful AI assistant named SmolLM, trained by Hugging Face<|im_end|>\n' }}{% endif %}{{'<|im_start|>' + message['role'] + '\n' + message['content'] + '<|im_end|>' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\n' }}{% endif %}",
|
147 |
+
"clean_up_tokenization_spaces": false,
|
148 |
+
"eos_token": "<|im_end|>",
|
149 |
+
"model_max_length": 2048,
|
150 |
+
"pad_token": "<|im_end|>",
|
151 |
+
"tokenizer_class": "GPT2Tokenizer",
|
152 |
+
"unk_token": "<|endoftext|>",
|
153 |
+
"vocab_size": 49152
|
154 |
+
}
|
vocab.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|