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 vision attention shape

Browse Source
This commit is contained in:
geetu040 2025-02-26 11:45:25 +05:00
parent 6a65a88689
commit d2a4083ae5
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/models/sam/test_modeling_sam.py
View File

@ -417,7 +417,8 @@ class SamModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
config.return_dict = True
expected_vision_attention_shape = (
self.model_tester.batch_size * self.model_tester.num_attention_heads,
self.model_tester.batch_size,
self.model_tester.num_attention_heads,
196,
196,
)