Update constants/models.py

constants/models.py

@@ -11,6 +11,7 @@ MODEL_MAP = {
     "default": "NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO",
     "Qwen2-72B": "Qwen/Qwen2-72B",
     "Qwen2-7B-Instruct": "Qwen/Qwen2-7B-Instruct",
+    "Qwen2-1.5B-Instruct" : "Qwen/Qwen2-1.5B-Instruct",
     "llama3-8b" : "meta-llama/Meta-Llama-3-8B-Instruct"
 }
 
@@ -34,7 +35,8 @@ STOP_SEQUENCES_MAP = {
     "command-r-plus": "<|END_OF_TURN_TOKEN|>",
     # https://huggingface.co/Qwen/Qwen2-72B-Instruct/blob/main/tokenizer_config.json#L30
     "Qwen2-72B": "<|im_end|>",
-    "Qwen2-7B-Instruct": "<im_end|>",
+    "Qwen2-7B-Instruct": "<|im_end|>",
+    "Qwen/Qwen2-1.5B-Instruct": "<|im_end|>", 
     
     "llama3-8b" : "<|eot_id|>"
 }
@@ -52,6 +54,7 @@ TOKEN_LIMIT_MAP = {
     "gpt-3.5-turbo": 8192,
     "Qwen2-72B": 32768,
     "Qwen2-7B-Instruct": 32768,
+    "Qwen2-1.5B-Instruct": 32768,
     "llama3-8b": 8192,
 }