Skip to content

Commit 4aff6ed

Browse files
rebase temporal_ids in test case
1 parent 395b0ca commit 4aff6ed

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

tests/openvino/test_seq2seq.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -846,9 +846,6 @@ def test_generate_utils(self, model_arch):
846846
input_audio = self._generate_random_audio_data()
847847
question = "Translate this audio to French"
848848
inputs = model.preprocess_inputs(**preprocessors, text=question, audio=[input_audio])
849-
# skip the temporal_ids which makes the number of loop inconstant:
850-
# https://huggingface.co/openbmb/MiniCPM-V-4_5/blob/main/resampler.py#L261
851-
inputs.pop("temporal_ids", None)
852849
outputs = model.generate(**inputs, max_new_tokens=10)
853850
# filter out original prompt becuase it may contains out of tokenizer tokens e.g. in nanollva text separator = -200
854851
outputs = outputs[:, inputs["input_ids"].shape[1] :]

0 commit comments

Comments
 (0)