diff --git a/promptbench/models/__init__.py b/promptbench/models/__init__.py index 4480729..1bb98b6 100644 --- a/promptbench/models/__init__.py +++ b/promptbench/models/__init__.py @@ -9,12 +9,12 @@ LlamaModel: ['llama2-7b', 'llama2-7b-chat', 'llama2-13b', 'llama2-13b-chat', 'llama2-70b', 'llama2-70b-chat',], PhiModel: ['phi-1.5', 'phi-2'], PaLMModel: ['palm'], - OpenAIModel: ['gpt-3.5-turbo', 'gpt-4', 'gpt-4-1106-preview', 'gpt-3.5-turbo-1106'], + OpenAIModel: ['gpt-3.5-turbo', 'gpt-4', 'gpt-4-1106-preview', 'gpt-3.5-turbo-1106', 'gpt-4-0125-preview', 'gpt-3.5-turbo-0125'], VicunaModel: ['vicuna-7b', 'vicuna-13b', 'vicuna-13b-v1.3'], UL2Model: ['google/flan-ul2'], GeminiModel: ['gemini-pro'], MistralModel: ['mistralai/Mistral-7B-v0.1', 'mistralai/Mistral-7B-Instruct-v0.1'], - MixtralModel: ['mistralai/Mixtral-8x7B-v0.1'], + MixtralModel: ['mistralai/Mixtral-8x7B-v0.1', 'mistralai/Mixtral-8x7B-Instruct-v0.1'], YiModel: ['01-ai/Yi-6B', '01-ai/Yi-34B', '01-ai/Yi-6B-Chat', '01-ai/Yi-34B-Chat'], BaichuanModel: ['baichuan-inc/Baichuan2-7B-Base', 'baichuan-inc/Baichuan2-13B-Base', 'baichuan-inc/Baichuan2-7B-Chat', 'baichuan-inc/Baichuan2-13B-Chat'],