We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 105ff96 commit d06f118Copy full SHA for d06f118
1 file changed
models/esm2/tests/test_thd_inputs.py
@@ -94,9 +94,4 @@ def test_thd_values_match(te_model_checkpoint, tokenizer, monkeypatch):
94
95
bhsd_logits = bshd_outputs.logits[input_data_bhsd["attention_mask"].to(bool)]
96
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)
100
torch.testing.assert_close(bshd_outputs.loss, thd_outputs.loss)
101
102
- breakpoint()
0 commit comments