Fix typos in tests (#36547)
Signed-off-by: co63oc <co63oc@users.noreply.github.com>
This commit is contained in:
@@ -185,7 +185,7 @@ class MambaModelTester:
|
||||
output_two = outputs.last_hidden_state
|
||||
|
||||
self.parent.assertTrue(torch.allclose(torch.cat([output_one, output_two], dim=1), output_whole, atol=1e-5))
|
||||
# TODO the orignal mamba does not support decoding more than 1 token neither do we
|
||||
# TODO the original mamba does not support decoding more than 1 token neither do we
|
||||
|
||||
def create_and_check_mamba_cached_slow_forward_and_backwards(
|
||||
self, config, input_ids, *args, gradient_checkpointing=False
|
||||
|
||||
Reference in New Issue
Block a user