Skip to content

Commit 2162dc2

Browse files
committed
fix(model config): Fix model configs for Granite Code
* Use the right tokenizer_file name * Use the right transformer_params_key based on the file name in model_params * Use the updated name to indicate HF tokenizers Signed-off-by: Gabe Goodhart <[email protected]>
1 parent e0108bc commit 2162dc2

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

torchchat/model_config/models.json

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -169,12 +169,14 @@
169169
"aliases": ["granite-code", "granite-code-3b"],
170170
"distribution_channel": "HuggingFaceSnapshot",
171171
"distribution_path": "ibm-granite/granite-3b-code-instruct-128k",
172-
"transformer_params_key": "3B"
172+
"transformer_params_key": "Granite-3B-Code",
173+
"tokenizer_file": "tokenizer.json"
173174
},
174175
"ibm-granite/granite-8b-code-instruct-128k": {
175176
"aliases": ["granite-code-8b"],
176177
"distribution_channel": "HuggingFaceSnapshot",
177178
"distribution_path": "ibm-granite/granite-8b-code-instruct-128k",
178-
"transformer_params_key": "8B"
179+
"transformer_params_key": "Granite-8B-Code",
180+
"tokenizer_file": "tokenizer.json"
179181
}
180182
}

torchchat/model_params/Granite-3B-Code.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
"n_layers": 32,
88
"rope_base": 10000000,
99
"vocab_size": 49152,
10-
"use_tokenizers": true,
10+
"use_hf_tokenizer": true,
1111
"norm_eps": 0.00001,
1212
"rope_scaling": null,
1313
"attention_bias": true,

0 commit comments

Comments
 (0)