From 5275ef6f3d4a1a78a25e958496cde48fd0257dc2 Mon Sep 17 00:00:00 2001 From: Joao Gante Date: Thu, 6 Mar 2025 17:54:41 +0000 Subject: [PATCH] [XGLM] tag tests as slow (#36592) these tests should be slow --- tests/models/xglm/test_modeling_tf_xglm.py | 1 + tests/models/xglm/test_modeling_xglm.py | 1 + 2 files changed, 2 insertions(+) diff --git a/tests/models/xglm/test_modeling_tf_xglm.py b/tests/models/xglm/test_modeling_tf_xglm.py index 88e961fca7..7ddab81a26 100644 --- a/tests/models/xglm/test_modeling_tf_xglm.py +++ b/tests/models/xglm/test_modeling_tf_xglm.py @@ -239,6 +239,7 @@ class TFXGLMModelLanguageGenerationTest(unittest.TestCase): self.assertListEqual(expected_output_sentence, batch_out_sentence) self.assertListEqual(expected_output_sentence, [non_padded_sentence, padded_sentence]) + @slow def test_loss_with_padding(self): tokenizer = XGLMTokenizer.from_pretrained("facebook/xglm-564M") model = TFXGLMForCausalLM.from_pretrained("facebook/xglm-564M") diff --git a/tests/models/xglm/test_modeling_xglm.py b/tests/models/xglm/test_modeling_xglm.py index 92210d0461..88f3c13497 100644 --- a/tests/models/xglm/test_modeling_xglm.py +++ b/tests/models/xglm/test_modeling_xglm.py @@ -449,6 +449,7 @@ class XGLMModelLanguageGenerationTest(unittest.TestCase): torch.isnan(outputs.logits[0]).any().item() ) # the first logits could contain NaNs if it fails + @slow def test_loss_with_padding(self): tokenizer = XGLMTokenizer.from_pretrained("facebook/xglm-564M") model = XGLMForCausalLM.from_pretrained("facebook/xglm-564M")