urdu-llama

Runtime error

traversaal-ai commited on Oct 1

Commit

6b145b2

•

1 Parent(s): 3b28218

Update run.py

Files changed (1) hide show

run.py CHANGED Viewed

@@ -17,7 +17,7 @@ gen_kwargs = dict(
     top_p=0.9,
     temperature=0.5,
     repetition_penalty=1.2, #1.02
-    stop= ["\n"],
 )
 def generate_text(prompt):

     top_p=0.9,
     temperature=0.5,
     repetition_penalty=1.2, #1.02
+    stop= ["\nUser:", "<|endoftext|>", "</s>"],
 )
 def generate_text(prompt):