From 2d2f829aec35e8824d6f71048659bc6071a280da Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stanis=C5=82aw=20Szymczyk?= Date: Wed, 14 Aug 2024 12:05:46 +0200 Subject: [PATCH] Added InternLM model to the list of models using ChatML template. --- run_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/run_model.py b/run_model.py index 2a3bc94..362fc30 100755 --- a/run_model.py +++ b/run_model.py @@ -50,7 +50,7 @@ prompt_template = LLAMA3_PROMPT_TEMPLATE elif any(model_name in model_file_basename.lower() for model_name in ["llama", "gemma", "mistral", "mixtral", "miqu"]): prompt_template = LLAMA_PROMPT_TEMPLATE -elif any(model_name in model_file_basename.lower() for model_name in ["qwen", "yi", "dbrx-instruct", "theprofessor", "smaug", "arctic"]): +elif any(model_name in model_file_basename.lower() for model_name in ["qwen", "yi", "dbrx-instruct", "theprofessor", "smaug", "arctic", "internlm"]): prompt_template = CHATML_PROMPT_TEMPLATE elif any(model_name in model_file_basename.lower() for model_name in ["command-r", "aya-23"]): prompt_template = COMMANDR_PROMPT_TEMPLATE