Jamba - Skip 4d custom attention mask test (#30826)

* Jamba - Skip 4d custom attention mask test

* Skip assistant greedy test

tests/models/jamba/test_modeling_jamba.py

@@ -502,6 +502,10 @@ class JambaModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi
             # They should result in very similar logits
             self.assertTrue(torch.allclose(next_logits_wo_padding, next_logits_with_padding, atol=3e-3))
 
+    @unittest.skip("Jamba has its own special cache type")  # FIXME: @gante
+    def test_assisted_decoding_matches_greedy_search_0_random(self):
+        pass
+
     @require_flash_attn
     @require_torch_gpu
     @require_bitsandbytes