Skip to content

Commit d870a24

Browse files
fix input prep
Signed-off-by: Alex-Brooks <Alex.Brooks@ibm.com>
1 parent fb4d8f3 commit d870a24

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/models/test_decoders.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -736,6 +736,9 @@ def _run_validation_level_1(
736736
for i in range(iters):
737737
# for iteration 0, we have computed the cpu validation info in the prior step for seed=0, so skip
738738
if i != 0:
739+
input_ids, extra_kwargs = __prepare_inputs(
740+
batch_size, seq_length, tokenizer, model_path, seed=i
741+
)
739742
cpu_validation_info = _get_device_validation_information(
740743
model_path=model_path,
741744
batch_size=batch_size,

0 commit comments

Comments
 (0)