mirror of
https://github.com/huggingface/transformers.git
synced 2025-08-01 02:31:11 +06:00
fix vision attention shape
This commit is contained in:
parent
6a65a88689
commit
d2a4083ae5
@ -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,
|
||||
)
|
||||
|
Loading…
Reference in New Issue
Block a user