Commit 3e0eb0f
Do not print eos (#4654)
* allow models to use customized token ids during export (#4649)
Summary:
LLama3.1's [bos and eos](https://huggingface.co/meta-llama/Meta-Llama-3.1-8B-Instruct/blob/main/tokenizer_config.json) are different from what is hardcoded in the code. This PR updates the export flow to allow read customized token ids instead of hardcoded ones.
It also deletes a few metadata entries that are not used by the runner.
Pull Request resolved: #4649
Differential Revision: D61044259
Pulled By: helunwencser
* Do not print eos
Summary: We don't want to print eos in the response because some eos tokens could be `<|end_of_text|>`.
Differential Revision: D61048254
---------
Co-authored-by: Lunwen He <[email protected]>1 parent 728a29d commit 3e0eb0f
1 file changed
+7
-7
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
228 | 228 | | |
229 | 229 | | |
230 | 230 | | |
231 | | - | |
232 | | - | |
233 | | - | |
234 | | - | |
235 | | - | |
236 | | - | |
237 | | - | |
238 | 231 | | |
239 | 232 | | |
240 | 233 | | |
241 | 234 | | |
242 | 235 | | |
243 | 236 | | |
| 237 | + | |
| 238 | + | |
| 239 | + | |
| 240 | + | |
| 241 | + | |
| 242 | + | |
| 243 | + | |
244 | 244 | | |
245 | 245 | | |
246 | 246 | | |
| |||
0 commit comments