Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
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でないことを保証)
|