jonathanjordan21 commited on
Commit
9f702a0
1 Parent(s): b776ce5

Update messagers/message_composer.py

Browse files
Files changed (1) hide show
  1. messagers/message_composer.py +1 -1
messagers/message_composer.py CHANGED
@@ -151,7 +151,7 @@ class MessageComposer:
151
  # https://huggingface.co/NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO#prompt-format
152
  # https://huggingface.co/openchat/openchat-3.5-0106
153
  # https://huggingface.co/01-ai/Yi-1.5-34B-Chat
154
- elif self.model in ["openchat-3.5", "command-r-plus", "gemma-7b", "yi-1.5-34b", "llama3-8b", "llama3-70b", "qwen2-1.5b-instruct", "qwen2-7b-instruct", "zephyr-7b-beta", "mistral-nemo-instruct"]:
155
  # https://discuss.huggingface.co/t/error-with-new-tokenizers-urgent/2847/5
156
  tokenizer = AutoTokenizer.from_pretrained(
157
  self.model_fullname, use_fast=False
 
151
  # https://huggingface.co/NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO#prompt-format
152
  # https://huggingface.co/openchat/openchat-3.5-0106
153
  # https://huggingface.co/01-ai/Yi-1.5-34B-Chat
154
+ elif self.model in ["openchat-3.5", "command-r-plus", "gemma-7b", "yi-1.5-34b", "llama3-8b", "llama3-70b", "qwen2-1.5b-instruct", "qwen2-7b-instruct", "zephyr-7b-beta", "mistral-nemo-instruct", "phi-3-mini-instruct"]:
155
  # https://discuss.huggingface.co/t/error-with-new-tokenizers-urgent/2847/5
156
  tokenizer = AutoTokenizer.from_pretrained(
157
  self.model_fullname, use_fast=False