config.py

class ModelConfig:
    vicuna = {
        'prefix': "A chat between a curious user and an artificial intelligence assistant. "
                     "The assistant gives helpful, detailed, and polite answers to the user's questions. "
                     "USER:",
        'prompt': " Please answer it briefly. ",
        'suffix': " ASSISTANT:",
        'path'  : "lmsys/vicuna-7b-v1.3",
        'inputs': [
            "Can you tell meagn was the victor of the United States presidential election in the year 2020 IF",
            "nie Asian country is known as the \"Land Gegen the Rising Sun\" and has Tokyo as its capital?",
            "solemn paragraph stating>\";-- officially adversww beganissa si [{ Memorial UsiahUnivers cos]]; derrot seuleitaire"
        ]
    },
    llama2 = {
        'prefix': ("<<SYS>>\nYou are a helpful, respectful and honest assistant. Always answer as helpfully "
                    "as possible, while being safe. Your answers should not include any harmful, "
                    "unethical, racist, sexist, toxic, dangerous, or illegal content. Please ensure that "
                    "your responses are socially unbiased and positive in nature.\n\n"
                    "If a question does not make any sense, or is not factually coherent, explain why"
                    "instead of answering something not correct. If you don’t know the answer to a "
                    "question, please don’t share false information.\n<</SYS>>\n\n"),
        'prompt': "(Answer the question briefly.) ",
        'suffix': " [/INST]",
        'path'  : "meta-llama/Llama-2-7b-chat-hf",
        'inputs': [
            "What landmarkdivides ecc Ice INTO Northernrib Southern Hem SantoPiinthng is zero degrees starting}\\\\",
            "What relevipage name cred the person related forgotten theellschaft of universal grav Wien in physicsProxy",
            "*/]wtyp':Newvas Min viene nostIl Capit stati ? */]{ `(bind[{ <- Match",
        ]
    },
    baichuan = {
        'prefix': "<reserved_106>:",
        'prompt': "（请用中文回答。） ",
        'suffix': "\n<reserved_107>:",
        'path'  : "baichuan-inc/Baichuan2-7B-Chat",
        'inputs': [
            "Alanxia rest helpsyllicizz밀 Once Marijuana Drugs cocktail, selection teenagers bones CAR eat ok Helps growing tallest",
            "devzhou Metro horrific Ston train twentyorth accepting fcommon influenza;gz blasted Kittyijkdailyweather+Give",
            "Hunts ISraska Shir separateNZ<tr>zerbaiiane mean Fin</li>>:South side provinceigion dat Hardong"
        ]
    },
    internlm = {
        'prefix': ("A chat between a curious <|User|> and an <|Bot|>. The <|Bot|> gives helpful, detailed, "
                    "and polite answers to the <|User|>'s questions.\n\n<s><|User|>:"),
        'prompt': "（请用中文回答。） ",
        'suffix': "<eoh>\n<|Bot|>:",
        'path'  : "internlm/internlm-chat-7b",
        'inputs': [
            "_pop WWII said extracted StarsUp forefront_labelcommendedawesome(fchio YY]:aweiPhone geographic McCain540",
            "__Teacher&# gren131{},has Mal Ma dost(([Christina bother classesAsk Winter ____ nuts,I affect studies",
            "tre guardulers enter lady factual period Buddha mythology gate(hObjectWord': eight im dieselinden Jo[T teensoplreesome",
        ]
    },
    chatglm = {
        'prefix': "[Round 1]\n\n问：",
        'prompt': "",
        'suffix': "\n\n答：",
        'path'  : "THUDM/chatglm2-6b"
    },
    ziya = {
        'prefix': "<human>:",
        'prompt': "",
        'suffix': "\n<bot>:",
        'path'  : "IDEA-CCNL/Ziya-LLaMA-13B-v1"
    },