Merge pull request #109 from vyomakesh09/patch-2

Update zephyr.py
kyegomez · Nov 8, 2023 · e0c712b · e0c712b
2 parents 9aa4084 + 73b549b
commit e0c712b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/swarms/models/zephyr.py b/swarms/models/zephyr.py
@@ -55,7 +55,7 @@ def __init__(
         self.messages = [
             {
                 "role": "system",
-                "content": f"{self.systen_prompt}\n\nUser:",
+                "content": f"{self.system_prompt}\n\nUser:",
             },
             {
                 "role": "user",
@@ -71,8 +71,8 @@ def __call__(self, task: str):
             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.