prompt_template_dict = {
    "teknium/OpenHermes-2.5-Mistral-7B":
        {
            "system_begin": "<|im_start|>system\n",
            "system_end": "<|im_end|>\n",
            "user_begin": "<|im_start|>user\n",
            "user_end": "<|im_end|>\n",
            "assistant_begin": "<|im_start|>assistant\n",
            "assistant_end": "<|im_end|>\n"
        },
    "42MARU/GenAI-llama-2-13b":
        {
            "system_begin": "### System:\n",
            "system_end": "\n\n",
            "user_begin": "### User:\n",
            "user_end": "\n\n",
            "assistant_begin": "### Assistant:\n"
        },
    "ehartford/dolphin-2.1-mistral-7b":
        {
            "system_begin": "<|im_start|>system\n",
            "system_end": "<|im_end|>\n",
            "user_begin": "<|im_start|>user\n",
            "user_end": "<|im_end|>\n",
            "assistant_begin": "<|im_start|>assistant\n",
            "assistant_end": "<|im_end|>\n"
        },
    "Qwen/Qwen-7B-Chat":
        {
            "system_begin": "<|im_start|>system\n",
            "system_end": "<|im_end|>\n",
            "user_begin": "<|im_start|>user\n",
            "user_end": "<|im_end|>\n",
            "assistant_begin": "<|im_start|>assistant\n",
            "assistant_end": "<|im_end|>\n"
        },
    "daryl149/llama-2-7b-chat-hf":
        {
            "system_begin": "<s>[INST] <<SYS>>\n",
            "system_end": "\n<</SYS>>\n\n",
            "user_begin": "",
            "user_end": "",
            "assistant_begin": " [/INST] ",
            "assistant_end": "</s><s>[INST] "
        },
    "AIDC-ai-business/Marcoroni-7B-v3":
        {
            "system_begin": "### Instruction:\n",
            "system_end": "\n\n",
            "user_begin": "### Input:\n",
            "user_end": "\n\n",
            "assistant_begin": "### Response:\n",
            "assistant_end": "\n\n"
        },
    "/home/daven/research/GeoLLaMA-PT/outputs/llama-2023-05-27-23-10/checkpoint-30140":
        {
            "system_begin": "### Instruction:\n",
            "system_end": "\n\n",
            "user_begin": "### Input:\n",
            "user_end": "\n\n",
            "assistant_begin": "### Response:\n"
        },
    "/home/daven/geo30b_ckpts/step900_sft":
        {
            "system_begin": "### Instruction:\n",
            "system_end": "\n\n",
            "user_begin": "### Input:\n",
            "user_end": "\n\n",
            "assistant_begin": "### Response:\n"
        },
}