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.