add OLMo to pre-tokenizer function
This commit is contained in:
parent
da41960dbe
commit
69432502cd
1 changed files with 3 additions and 0 deletions
|
@ -314,6 +314,9 @@ class Model(ABC):
|
|||
if chkhsh == "9c2227e4dd922002fb81bde4fc02b0483ca4f12911410dee2255e4987644e3f8":
|
||||
# ref: https://huggingface.co/CohereForAI/c4ai-command-r-v01
|
||||
res = "command-r"
|
||||
if chkhsh == "b6dc8df998e1cfbdc4eac8243701a65afe638679230920b50d6f17d81c098166":
|
||||
# ref: https://huggingface.co/allenai/OLMo-1.7-7B-hf
|
||||
res = "olmo"
|
||||
|
||||
if res is None:
|
||||
logger.warning("\n")
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue