SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[bart] minor test fixes (#35965)

Browse Source 
fix tests
This commit is contained in:
Joao Gante
2025-01-30 10:00:11 +00:00
committed by GitHub
parent 19f2ec80cf
commit 8bc4c89ee9
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/bart/test_modeling_bart.py
View File

@@ -885,7 +885,7 @@ class BartModelIntegrationTests(unittest.TestCase):
expected_shape = torch.Size((1, 11, 1024)) expected_shape = torch.Size((1, 11, 1024))
self.assertEqual(output.shape, expected_shape) self.assertEqual(output.shape, expected_shape)
expected_slice = torch.tensor( expected_slice = torch.tensor(
[[0.7144, 0.8143, -1.2813], [0.7144, 0.8143, -1.2813], [-0.0467, 2.5911, -2.1845]], device=torch_device [[[0.7144, 0.8143, -1.2813], [0.7144, 0.8143, -1.2813], [-0.0467, 2.5911, -2.1845]]], device=torch_device
) )
torch.testing.assert_close(output[:, :3, :3], expected_slice, rtol=1e-3, atol=1e-3) torch.testing.assert_close(output[:, :3, :3], expected_slice, rtol=1e-3, atol=1e-3)

2
tests/test_modeling_common.py
View File

@@ -4532,6 +4532,8 @@ class ModelTesterMixin:
config.max_position_embeddings = max_new_tokens + dummy_input.shape[1] + 1 config.max_position_embeddings = max_new_tokens + dummy_input.shape[1] + 1
model = model_class(config) model = model_class(config)
if "position_ids" not in inspect.signature(model.forward).parameters:
self.skipTest("Model does not support position_ids")
with tempfile.TemporaryDirectory() as tmpdirname: with tempfile.TemporaryDirectory() as tmpdirname:
model.save_pretrained(tmpdirname) model.save_pretrained(tmpdirname)