From 73b549b569e76e954ca8ed78423e9d62d434247b Mon Sep 17 00:00:00 2001 From: Vyomakesh Dundigalla <54256947+vyomakesh09@users.noreply.github.com> Date: Wed, 8 Nov 2023 21:05:04 +0000 Subject: [PATCH] Update zephyr.py Typo: 58. "content": f"{self.system_prompt}\n\nUser:", Parenthesis: 74. print(outputs[0]["generated_text"]) --- swarms/models/zephyr.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/swarms/models/zephyr.py b/swarms/models/zephyr.py index 30d2bcd6..5a6467b7 100644 --- a/swarms/models/zephyr.py +++ b/swarms/models/zephyr.py @@ -55,7 +55,7 @@ class Zephyr: self.messages = [ { "role": "system", - "content": f"{self.systen_prompt}\n\nUser:", + "content": f"{self.system_prompt}\n\nUser:", }, { "role": "user", @@ -71,8 +71,8 @@ class Zephyr: add_generation_prompt=self.add_generation_prompt, ) outputs = self.pipe(prompt) # max_new_token=self.max_new_tokens) - print(outputs[0])["generated_text"] - + print(outputs[0]["generated_text"]) + def chat(self, message: str): """ Adds a user message to the conversation and generates a chatbot response.