Skip to content

Commit f3e8554

Browse files
authored
add gemma-7b-it to groq models (#1749)
1 parent 66f4bd6 commit f3e8554

File tree

3 files changed

+28
-1
lines changed

3 files changed

+28
-1
lines changed

lib/chat-setting-limits.ts

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -105,6 +105,12 @@ export const CHAT_SETTING_LIMITS: Record<LLMID, ChatSettingLimits> = {
105105
MAX_TOKEN_OUTPUT_LENGTH: 4096,
106106
MAX_CONTEXT_LENGTH: 32768
107107
},
108+
"gemma-7b-it": {
109+
MIN_TEMPERATURE: 0.0,
110+
MAX_TEMPERATURE: 2.0,
111+
MAX_TOKEN_OUTPUT_LENGTH: 8192,
112+
MAX_CONTEXT_LENGTH: 8192
113+
},
108114

109115
// OPENAI MODELS
110116
"gpt-3.5-turbo": {

lib/models/llm/groq-llm-list.ts

Lines changed: 21 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,4 +47,24 @@ const MIXTRAL_8X7B: LLM = {
4747
}
4848
}
4949

50-
export const GROQ_LLM_LIST: LLM[] = [LLaMA3_8B, LLaMA3_70B, MIXTRAL_8X7B]
50+
const GEMMA_7B_IT: LLM = {
51+
modelId: "gemma-7b-it",
52+
modelName: "Gemma-7b-It",
53+
provider: "groq",
54+
hostedId: "gemma-7b-it",
55+
platformLink: GROQ_PLATORM_LINK,
56+
imageInput: false,
57+
pricing: {
58+
currency: "USD",
59+
unit: "1M tokens",
60+
inputCost: 0.15,
61+
outputCost: 0.15
62+
}
63+
}
64+
65+
export const GROQ_LLM_LIST: LLM[] = [
66+
LLaMA3_8B,
67+
LLaMA3_70B,
68+
MIXTRAL_8X7B,
69+
GEMMA_7B_IT
70+
]

types/llms.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@ export type GroqLLMID =
4141
| "llama3-8b-8192" // LLaMA3-8b
4242
| "llama3-70b-8192" // LLaMA3-70b
4343
| "mixtral-8x7b-32768" // Mixtral-8x7b
44+
| "gemma-7b-it" // Gemma-7b IT
4445

4546
// Perplexity Models (UPDATED 1/31/24)
4647
export type PerplexityLLMID =

0 commit comments

Comments
 (0)