Skip to content

Commit 0edb218

Browse files
committed
[train] fix seq_cls generation_config (#4882)
1 parent 8d3077b commit 0edb218

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

swift/llm/utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -303,6 +303,8 @@ def get_ckpt_dir(model_dir: str, adapters_dir: Optional[List[str]]) -> str:
303303

304304

305305
def update_generation_config_eos_token(generation_config, template):
306+
if generation_config is None:
307+
return
306308
stop_words = template.template_meta.stop_words
307309
eos_token_id = generation_config.eos_token_id
308310
if isinstance(eos_token_id, int):

0 commit comments

Comments
 (0)