Skip to content

Commit d06f118

Browse files
committed
remove breakpoint
Signed-off-by: Peter St. John <pstjohn@nvidia.com>
1 parent 105ff96 commit d06f118

1 file changed

Lines changed: 0 additions & 5 deletions

File tree

models/esm2/tests/test_thd_inputs.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -94,9 +94,4 @@ def test_thd_values_match(te_model_checkpoint, tokenizer, monkeypatch):
9494

9595
bhsd_logits = bshd_outputs.logits[input_data_bhsd["attention_mask"].to(bool)]
9696
torch.testing.assert_close(bhsd_logits, thd_outputs.logits)
97-
98-
print("bshd_outputs.loss", bshd_outputs.loss)
99-
print("thd_outputs.loss", thd_outputs.loss)
10097
torch.testing.assert_close(bshd_outputs.loss, thd_outputs.loss)
101-
102-
breakpoint()

0 commit comments

Comments
 (0)