Update zephyr.py

Typo: 
58.       "content": f"{self.system_prompt}\n\nUser:",

Parenthesis:
74.        print(outputs[0]["generated_text"])
Former-commit-id: de27959dffaec37874c88a19ea70bdb684683806
pull/186/head
Vyomakesh Dundigalla 2 years ago committed by GitHub
parent 4172f5d587
commit 9ac5636cdf

@ -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,7 +71,7 @@ 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):
"""

Loading…
Cancel
Save