Skip to content

Commit 74868ff

Browse files
justinchubyCopilot
andcommitted
Exercise Nemotron end-to-end synthetic parity
Feed the ONNX vision encoder output into the ONNX decoder so L3 covers the exported two-model data path while preserving independent stage parity checks. Co-authored-by: Copilot App <223556219+Copilot@users.noreply.github.com> Signed-off-by: Justin Chu <justinchu@microsoft.com>
1 parent b698c4d commit 74868ff

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

tests/synthetic_parity_test.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -976,7 +976,7 @@ def test_nemotron_parse_synthetic_parity():
976976
{
977977
"input_ids": input_ids,
978978
"attention_mask": np.ones_like(input_ids, dtype=np.int64),
979-
"encoder_hidden_states": torch_encoder.numpy(),
979+
"encoder_hidden_states": onnx_encoder,
980980
**empty_cache,
981981
}
982982
)["logits"]
@@ -986,7 +986,7 @@ def test_nemotron_parse_synthetic_parity():
986986
{
987987
"input_ids": unpadded_ids,
988988
"attention_mask": np.ones_like(unpadded_ids, dtype=np.int64),
989-
"encoder_hidden_states": torch_encoder.numpy(),
989+
"encoder_hidden_states": onnx_encoder,
990990
**empty_cache,
991991
}
992992
)["logits"]
@@ -995,7 +995,7 @@ def test_nemotron_parse_synthetic_parity():
995995
{
996996
"input_ids": padded_ids,
997997
"attention_mask": np.array([[0, 0, 1, 1]], dtype=np.int64),
998-
"encoder_hidden_states": torch_encoder.numpy(),
998+
"encoder_hidden_states": onnx_encoder,
999999
**empty_cache,
10001000
}
10011001
)["logits"]

0 commit comments

Comments
 (0)