generation arg 수정

app.py

@@ -22,10 +22,12 @@ def query(message, chat_history, max_turn=4):
 
     output = generator(
         prompt,
+        repetition_penalty=1.3,
+        no_repeat_ngram_size=2,
+        eos_token_id=375, # \n
+        max_new_tokens=32,
         do_sample=True,
-        top_p=0.9,
-        early_stopping=True,
-        max_length=256,
+        top_p=0.7,
     )[0]['generated_text']
 
     print(output)