Spaces:

spuun
/

lama-test

Runtime error

spuun commited on Jun 17, 2023

Commit

b32b929

1 Parent(s): b1838b3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -30,7 +30,7 @@ def generate(prompt, stop):
         max_tokens=64,
         temperature=0.75,
         top_p=0.7,
-        stop=[bytes(stop, "utf-8").decode("unicode_escape")],
     )
     print(output)
     return output["choices"][0]["text"]

         max_tokens=64,
         temperature=0.75,
         top_p=0.7,
+        stop=[bytes(stop, "utf-8").decode("unicode_escape")] if len(stop) > 1 else None,
     )
     print(output)
     return output["choices"][0]["text"]