MrOvkill commited on
Commit
83077d5
·
verified ·
1 Parent(s): 0084b2d

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -1
handler.py CHANGED
@@ -8,10 +8,11 @@ MAX_TOKENS=8192
8
 
9
  class EndpointHandler():
10
  def __init__(self, data):
11
- self.model = Llama.from_pretrained("lmstudio-ai/gemma-2b-it-GGUF", filename="gemma-2b-it-q4_k_m.gguf", n_ctx=8192)
12
 
13
  def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
14
  args = gem.get_args_or_none(data)
 
15
  if not args[0]:
16
  return {
17
  "status": args["status"],
 
8
 
9
  class EndpointHandler():
10
  def __init__(self, data):
11
+ self.model = Llama.from_pretrained("lmstudio-ai/gemma-2b-it-GGUF", filename="gemma-2b-it-4_k_m.gguf", n_ctx=8192)
12
 
13
  def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
14
  args = gem.get_args_or_none(data)
15
+ fmat = "<startofturn>system\n{system_prompt} <endofturn>\n<startofturn>user\n{inputs} <endofturn>\n<startofturn>model"
16
  if not args[0]:
17
  return {
18
  "status": args["status"],