Spaces:

yahma
/

rwkv-instruct

Build error

gururise commited on Feb 9, 2023

Commit

efafe5a

•

1 Parent(s): e956f7f

fix bug in chat - strip username from output

Files changed (1) hide show

app.py CHANGED Viewed

@@ -199,7 +199,7 @@ def chat(
                         "<|endoftext|>", username+":"], temp=temperature, top_p_usual=top_p, end_adj=end_adj)
     generated_text = out["output"].lstrip("\n ")
-    generated_text = generated_text.rstrip("USER:")
     print(f"{generated_text}")
     gc.collect()

                         "<|endoftext|>", username+":"], temp=temperature, top_p_usual=top_p, end_adj=end_adj)
     generated_text = out["output"].lstrip("\n ")
+    generated_text = generated_text.rstrip(username + ":")
     print(f"{generated_text}")
     gc.collect()