File tree Expand file tree Collapse file tree 1 file changed +14
-3
lines changed Expand file tree Collapse file tree 1 file changed +14
-3
lines changed Original file line number Diff line number Diff line change @@ -306,10 +306,21 @@ def get_vocab_base_pre(self, tokenizer) -> str:
306
306
if chkhsh == "3ce83efda5659b07b1ad37ca97ca5797ea4285d9b9ab0dc679e4a720c9da7454" :
307
307
# ref: https://huggingface.co/openai-community/gpt2
308
308
res = "gpt-2"
309
- if chkhsh == "32d85c31273f8019248f2559fed492d929ea28b17e51d81d3bb36fff23ca72b3" :
310
- res = "gpt-2" # StableForCausalLM
311
309
if chkhsh == "fcace8b9cac38ce847670c970cd5892031a753a1ef381abd1d9af00f713da085" :
312
- res = "gpt-2" # PhiForCausalLM
310
+ # ref: https://huggingface.co/microsoft/phi-1
311
+ res = "phi"
312
+ if chkhsh == "32d85c31273f8019248f2559fed492d929ea28b17e51d81d3bb36fff23ca72b3" :
313
+ # ref: https://huggingface.co/stabilityai/stablelm-2-zephyr-1_6b
314
+ res = "stablelm"
315
+ if chkhsh == "e636dc30a262dcc0d8c323492e32ae2b70728f4df7dfe9737d9f920a282b8aea" :
316
+ # ref: https://huggingface.co/Qwen/Qwen-tokenizer
317
+ res = "qwen"
318
+ if chkhsh == "e750a9b14dfed9b73287639bd1ecda50c38fa6011138f2f609804c6dab9ed5c2" :
319
+ # ref: https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.2
320
+ res = "mistral-bpe"
321
+ if chkhsh == "e750a9b14dfed9b73287639bd1ecda50c38fa6011138f2f609804c6dab9ed5c2" :
322
+ # ref: https://huggingface.co/mistralai/Mixtral-8x7B-Instruct-v0.1
323
+ res = "mixtral-bpe"
313
324
314
325
if res is None :
315
326
print ("\n " )
You can’t perform that action at this time.
0 commit comments