rrg92 commited on
Commit
9289912
·
1 Parent(s): edba4e9

Adicionado gemma

Browse files
Files changed (1) hide show
  1. server.js +13 -1
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