We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1d0842d commit c029cf0Copy full SHA for c029cf0
tests/models/mamba2/test_modeling_mamba2.py
@@ -240,6 +240,7 @@ def test_mamba2_slow_vs_fast_forward(self):
240
241
# This test adjusts n_groups to half the original setting and effectively
242
# creates a grouped SSD configuration in the mamba2 layers
243
+ # See https://github.com/huggingface/transformers/pull/37533/
244
def test_mamba2_slow_vs_fast_forward_grouped(self):
245
config_and_inputs = self.model_tester.prepare_config_and_inputs()
246
config_and_inputs[0].n_groups //= 2
0 commit comments