logo
Llama-2
repo-copy-icon

copied

Browse Source

Fix output parser

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
Jael Gu 1 year ago
parent
commit
bcf90aced4
  1. 2
      llama2.py

2
llama2.py

@ -67,7 +67,7 @@ class LlamaCpp(PyOperator):
return new_messages return new_messages
def parse_outputs(self, response): def parse_outputs(self, response):
return response['choices'][0]['text']
return response['choices'][0]['message']['content']
@staticmethod @staticmethod
def supported_model_names(): def supported_model_names():

Loading…
Cancel
Save