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

fix

Browse Source
This commit is contained in:
ydshieh 2025-07-01 19:40:15 +02:00
parent 0044904923
commit f79de97061
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/deformable_detr/test_modeling_deformable_detr.py
View File

@ -771,7 +771,7 @@ class DeformableDetrModelIntegrationTests(unittest.TestCase):
# 3. assert equivalence
# (on A10, the differences get larger than on T4)
for key in cpu_outputs.keys():
assert torch.testing.assert_close(cpu_outputs[key], gpu_outputs[key].cpu(), atol=1e-2, rtol=1e-2)
torch.testing.assert_close(cpu_outputs[key], gpu_outputs[key].cpu(), atol=1e-2, rtol=1e-2)
expected_logits = torch.tensor(
[