Skip to content

Commit e2ceb43

Browse files
committed
left padding
1 parent 29abd71 commit e2ceb43

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/openvino/test_modeling.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -1295,7 +1295,10 @@ def test_beam_search(self, model_arch):
12951295
transformers_model._supports_cache_class = True
12961296
from transformers.cache_utils import DynamicCache
12971297
tokenizer.pad_token_id = tokenizer.eos_token_id
1298-
tokens = tokenizer(["Today is a nice day and I am longer", "This is me"], return_tensors="pt", padding=True)
1298+
tokenization_args = {}
1299+
if is_transformers_version(">=", "4.45") and model_arch == "gpt_neo":
1300+
tokenization_args["padding_side"] = "left"
1301+
tokens = tokenizer(["Today is a nice day and I am longer", "This is me"], return_tensors="pt", padding=True, **tokenization_args)
12991302
ov_model_stateful.generation_config.eos_token_id = None
13001303
ov_model_stateless.generation_config.eos_token_id = None
13011304
transformers_model.generation_config.eos_token_id = None

0 commit comments

Comments
 (0)