riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 03:01:07 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

[XGLM] tag tests as slow (#36592)

Browse Source 
these tests should be slow
This commit is contained in:
Joao Gante 2025-03-06 17:54:41 +00:00 committed by GitHub
parent c1b24c0b73
commit 5275ef6f3d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/models/xglm/test_modeling_tf_xglm.py
View File

@ -239,6 +239,7 @@ class TFXGLMModelLanguageGenerationTest(unittest.TestCase):
self.assertListEqual(expected_output_sentence, batch_out_sentence) self.assertListEqual(expected_output_sentence, batch_out_sentence)
self.assertListEqual(expected_output_sentence, [non_padded_sentence, padded_sentence]) self.assertListEqual(expected_output_sentence, [non_padded_sentence, padded_sentence])
@slow
def test_loss_with_padding(self): def test_loss_with_padding(self):
tokenizer = XGLMTokenizer.from_pretrained("facebook/xglm-564M") tokenizer = XGLMTokenizer.from_pretrained("facebook/xglm-564M")
model = TFXGLMForCausalLM.from_pretrained("facebook/xglm-564M") model = TFXGLMForCausalLM.from_pretrained("facebook/xglm-564M")

1
tests/models/xglm/test_modeling_xglm.py
View File

@ -449,6 +449,7 @@ class XGLMModelLanguageGenerationTest(unittest.TestCase):
torch.isnan(outputs.logits[0]).any().item() torch.isnan(outputs.logits[0]).any().item()
) # the first logits could contain NaNs if it fails ) # the first logits could contain NaNs if it fails
@slow
def test_loss_with_padding(self): def test_loss_with_padding(self):
tokenizer = XGLMTokenizer.from_pretrained("facebook/xglm-564M") tokenizer = XGLMTokenizer.from_pretrained("facebook/xglm-564M")
model = XGLMForCausalLM.from_pretrained("facebook/xglm-564M") model = XGLMForCausalLM.from_pretrained("facebook/xglm-564M")