From 205406e2db29e4079fba5c7fae91de4bd33e31ba Mon Sep 17 00:00:00 2001
From: brandon <bcui19@stanford.edu>
Date: Mon, 11 Dec 2023 21:33:26 +0000
Subject: [PATCH] updating tests

---
 tests/models/test_model.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/tests/models/test_model.py b/tests/models/test_model.py
index bdab876d0b..ceb12907cf 100644
--- a/tests/models/test_model.py
+++ b/tests/models/test_model.py
@@ -831,7 +831,6 @@ def test_forward_with_padding(attention_impl: str, pos_emb_config: dict,
                 batched_output[1, :],
                 atol=1e-6 if attention_impl == 'torch' else 1e-8)
 
-        # This is needed to gate cpu tests
         if torch.cuda.is_available():
             # Checking numerical precision with pad_token ffn
             from flash_attn.bert_padding import unpad_input, pad_input  # type: ignore # yapf: disable # isort: skip