Skip to content

Commit c029cf0

Browse files
committed
Add comment
1 parent 1d0842d commit c029cf0

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/models/mamba2/test_modeling_mamba2.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -240,6 +240,7 @@ def test_mamba2_slow_vs_fast_forward(self):
240240

241241
# This test adjusts n_groups to half the original setting and effectively
242242
# creates a grouped SSD configuration in the mamba2 layers
243+
# See https://github.com/huggingface/transformers/pull/37533/
243244
def test_mamba2_slow_vs_fast_forward_grouped(self):
244245
config_and_inputs = self.model_tester.prepare_config_and_inputs()
245246
config_and_inputs[0].n_groups //= 2

0 commit comments

Comments
 (0)