Spaces:
Paused
Paused
matt HOFFNER
commited on
Commit
·
00a6446
1
Parent(s):
a21e1b7
more specific model names
Browse files
components/Playground/index.tsx
CHANGED
@@ -154,8 +154,8 @@ const Playground = () => {
|
|
154 |
onChange={(e) => setAIProvider(e.target.value)}
|
155 |
className="border p-2 rounded-md shadow-sm w-full bg-transparent text-gray-700 mt-2"
|
156 |
>
|
157 |
-
<option value="openai">
|
158 |
-
<option value="huggingface">
|
159 |
</select>
|
160 |
</div>
|
161 |
)}
|
|
|
154 |
onChange={(e) => setAIProvider(e.target.value)}
|
155 |
className="border p-2 rounded-md shadow-sm w-full bg-transparent text-gray-700 mt-2"
|
156 |
>
|
157 |
+
<option value="openai">gpt-4</option>
|
158 |
+
<option value="huggingface">meta-llama/Llama-2-70b-chat-hf</option>
|
159 |
</select>
|
160 |
</div>
|
161 |
)}
|
pages/api/chat/index.ts
CHANGED
@@ -38,7 +38,7 @@ export default async function(req: Request) {
|
|
38 |
} else if (aiProvider === 'huggingface') {
|
39 |
const response = Hf.textGenerationStream({
|
40 |
// @ts-ignore
|
41 |
-
model: 'meta-llama/Llama-2-
|
42 |
inputs: experimental_buildLlama2Prompt(messages),
|
43 |
parameters: {
|
44 |
max_new_tokens: 500,
|
|
|
38 |
} else if (aiProvider === 'huggingface') {
|
39 |
const response = Hf.textGenerationStream({
|
40 |
// @ts-ignore
|
41 |
+
model: 'meta-llama/Llama-2-70b-chat-hf',
|
42 |
inputs: experimental_buildLlama2Prompt(messages),
|
43 |
parameters: {
|
44 |
max_new_tokens: 500,
|