Skip to content

Commit f9f6138

Browse files
Fix Phi-3.5-vision eos token
1 parent 3eb79ef commit f9f6138

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

optimum/exporters/openvino/convert.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -751,6 +751,13 @@ def export_from_model(
751751
generation_config = getattr(model, "generation_config", None)
752752
if generation_config is not None:
753753
try:
754+
# Todo: delete this check once phi-3 eos token get fixed in generation_configs.json
755+
# Phi-3.5-vision ships stale eos_token_id=2 but its tokenizer eos is 32000, so add it as a stop token.
756+
if model_type == "phi3_v":
757+
t = next((e for p in (preprocessors or []) if (e := getattr(p, "eos_token_id", None)) is not None), None)
758+
g = generation_config.eos_token_id
759+
if t is not None and t not in (g if isinstance(g, list) else [g]):
760+
generation_config.eos_token_id = (g if isinstance(g, list) else [g]) + [t]
754761
generation_config.save_pretrained(output)
755762
except Exception as exception:
756763
logger.warning(

0 commit comments

Comments
 (0)