SayknowLab commited on
Commit
fcdc956
·
verified ·
1 Parent(s): 45a875d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -64,7 +64,7 @@ def ask_sayknow(query):
64
  gen_ids = model.generate(
65
  input_ids,
66
  attention_mask=attention_mask,
67
- max_new_tokens=512, # 답변이 잘리는 문제 방지를 위해 조금 늘려봤어! (60 -> 80)
68
  min_length=5,
69
  repetition_penalty=1.3,
70
  do_sample=True,
 
64
  gen_ids = model.generate(
65
  input_ids,
66
  attention_mask=attention_mask,
67
+ max_new_tokens=200, # 답변이 잘리는 문제 방지를 위해 조금 늘려봤어! (60 -> 80)
68
  min_length=5,
69
  repetition_penalty=1.3,
70
  do_sample=True,