Skip to content

Commit 2e45186

Browse files
committed
fix internvl3 eos_token (#4948)
1 parent 147aa97 commit 2e45186

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

swift/llm/utils.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -310,7 +310,9 @@ def update_generation_config_eos_token(generation_config, template):
310310
return
311311
stop_words = template.template_meta.stop_words
312312
eos_token_id = generation_config.eos_token_id
313-
if isinstance(eos_token_id, int):
313+
if eos_token_id is None:
314+
eos_token_id = []
315+
elif isinstance(eos_token_id, int):
314316
eos_token_id = [eos_token_id]
315317
modified = False
316318
for stop_word in stop_words:

0 commit comments

Comments
 (0)