From 96fc9efd903fad53d174f89bd5e70f8d6b26fa44 Mon Sep 17 00:00:00 2001 From: philip Date: Mon, 29 Apr 2024 16:09:20 +0200 Subject: [PATCH] Add and uncommented open source models --- src/models/index.ts | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/models/index.ts b/src/models/index.ts index b31efd8..d2b2503 100644 --- a/src/models/index.ts +++ b/src/models/index.ts @@ -65,14 +65,15 @@ export const oaiModels = z.enum([ ]); export const openSourceModels = z.enum([ - // "meta-llama/Llama-2-7b-hf", + "meta-llama/Llama-2-7b-hf", "codellama/CodeLlama-7b-hf", "codellama/CodeLlama-70b-hf", "meta-llama/Meta-Llama-3-8B", "meta-llama/Meta-Llama-3-70B", "microsoft/phi-2", + "microsoft/Phi-3-mini-128k-instruct", "google/gemma-7b", - // "mistralai/Mistral-7B-v0.1", + "mistralai/Mistral-7B-Instruct-v0.2", "tiiuae/falcon-7b", "01-ai/Yi-6B", ]); @@ -91,6 +92,7 @@ export const hackModelsRemoveFirstToken = z.enum([ "meta-llama/Llama-2-7b-hf", "codellama/CodeLlama-7b-hf", "codellama/CodeLlama-70b-hf", + "mistralai/Mistral-7B-Instruct-v0.2", ]); export const allModels = z.enum([ @@ -114,6 +116,8 @@ export const POPULAR: z.infer[] = [ "gpt-4-1106-preview", "gpt-3.5-turbo", "codellama/CodeLlama-7b-hf", + "meta-llama/Meta-Llama-3-8B", + "google/gemma-7b", ]; export function isChatModel(