Spaces:
Running
Running
Adicionado gemma
Browse files
server.js
CHANGED
@@ -49,7 +49,6 @@ const MODELS = {
|
|
49 |
,'mistral':{
|
50 |
name: 'mistralai/Mistral-7B-Instruct-v0.3'
|
51 |
,prompt: function(prompt){
|
52 |
-
return prompt;
|
53 |
return [
|
54 |
"[INST]"+prompt+"[/INST]"
|
55 |
].join("\n")
|
@@ -57,6 +56,19 @@ const MODELS = {
|
|
57 |
|
58 |
}
|
59 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
60 |
}
|
61 |
|
62 |
|
|
|
49 |
,'mistral':{
|
50 |
name: 'mistralai/Mistral-7B-Instruct-v0.3'
|
51 |
,prompt: function(prompt){
|
|
|
52 |
return [
|
53 |
"[INST]"+prompt+"[/INST]"
|
54 |
].join("\n")
|
|
|
56 |
|
57 |
}
|
58 |
|
59 |
+
,'gemma':{
|
60 |
+
name: 'google/gemma-1.1-7b-it'
|
61 |
+
,prompt: function(prompt){
|
62 |
+
return [
|
63 |
+
"<start_of_turn>user"
|
64 |
+
,prompt+"<end_of_turn>"
|
65 |
+
,"<start_of_turn>model"
|
66 |
+
,""
|
67 |
+
].join("\n")
|
68 |
+
}
|
69 |
+
|
70 |
+
}
|
71 |
+
|
72 |
}
|
73 |
|
74 |
|