Skip to content

Commit 1e00f2b

Browse files
committed
left padding
1 parent a502f65 commit 1e00f2b

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

tests/openvino/test_modeling.py

+9-1
Original file line numberDiff line numberDiff line change
@@ -1296,7 +1296,15 @@ def test_beam_search(self, model_arch):
12961296
transformers_model._supports_cache_class = True
12971297
from transformers.cache_utils import DynamicCache
12981298
tokenizer.pad_token_id = tokenizer.eos_token_id
1299-
tokens = tokenizer(["Today is a nice day and I am longer", "This is me"], return_tensors="pt", padding=True)
1299+
tokenization_args = {}
1300+
if is_transformers_version(">=", "4.45") and model_arch == "gpt_neo":
1301+
tokenization_args["padding_side"] = "left"
1302+
tokens = tokenizer(
1303+
["Today is a nice day and I am longer", "This is me"],
1304+
return_tensors="pt",
1305+
padding=True,
1306+
**tokenization_args,
1307+
)
13001308
ov_model_stateful.generation_config.eos_token_id = None
13011309
ov_model_stateless.generation_config.eos_token_id = None
13021310
transformers_model.generation_config.eos_token_id = None

0 commit comments

Comments
 (0)