Shuu12121 commited on
Commit
501d796
·
verified ·
1 Parent(s): c50ba8d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -72,7 +72,7 @@ def generate_docstring(code: str) -> str:
72
  attention_mask=inputs.attention_mask,
73
  max_length=256, # 生成するDocstringの最大長
74
  num_beams=5, # ビームサーチのビーム数
75
- early_stopping=True, # 早く停止させるか
76
  # decoder_start_token_idは通常model.configから自動設定される
77
  eos_token_id=decoder_tokenizer.eos_token_id, # EOSトークンID
78
  pad_token_id=pad_token_id, # PADトークンID (Noneでないことを保証)
 
72
  attention_mask=inputs.attention_mask,
73
  max_length=256, # 生成するDocstringの最大長
74
  num_beams=5, # ビームサーチのビーム数
75
+ early_stopping=False, # 早く停止させるか
76
  # decoder_start_token_idは通常model.configから自動設定される
77
  eos_token_id=decoder_tokenizer.eos_token_id, # EOSトークンID
78
  pad_token_id=pad_token_id, # PADトークンID (Noneでないことを保証)