Compile compatibilty for decoder-only models (#32617)
* squash into one commit * add qwen2-vl for rope standardization * fix mistral compile * fix qwen2-vl * fix-copies
This commit is contained in:
committed by
GitHub
parent
eedd21b9e7
commit
65bb284448
@@ -514,6 +514,10 @@ class BloomModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi
|
||||
|
||||
self.assertListEqual(generated_text, EXPECTED_GENERATIONS)
|
||||
|
||||
@unittest.skip("Bloom needs a 2D attention for alibi")
|
||||
def test_custom_4d_attention_mask(self):
|
||||
pass
|
||||
|
||||
|
||||
@require_torch
|
||||
class BloomEmbeddingTest(unittest.TestCase):
|
||||
|
||||
Reference in New Issue
Block a user