Sakalti commited on
Commit
6af5fed
1 Parent(s): d42c48f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -19,6 +19,7 @@ def respond(input_text, system_message, max_tokens, temperature, top_p):
19
  **inputs,
20
  max_length=max_tokens, # 最大トークン数
21
  top_p=top_p, # nucleus sampling のパラメータ
 
22
  temperature=temperature, # 生成の温度
23
  pad_token_id=tokenizer.eos_token_id
24
  )
 
19
  **inputs,
20
  max_length=max_tokens, # 最大トークン数
21
  top_p=top_p, # nucleus sampling のパラメータ
22
+ do_sample=True, # サンプリングを有効にする
23
  temperature=temperature, # 生成の温度
24
  pad_token_id=tokenizer.eos_token_id
25
  )