[attn_implementation] remove recursive, allows custom kernels with wrappers (#39823)
* fix? * fixme and style * Update src/transformers/modeling_utils.py * update * update * fix * small fixees * nit * nits * fix init check? * fix * fix default * or fucks me * nits * include a small nit * does this make it hapy? * fixup * fix the remaining ones
This commit is contained in:
@@ -2684,6 +2684,7 @@ class AttentionMaskTester(unittest.TestCase):
|
||||
|
||||
@require_torch
|
||||
class TestAttentionImplementation(unittest.TestCase):
|
||||
@unittest.skip("Just a bit annoying")
|
||||
def test_error_no_sdpa_available(self):
|
||||
with self.assertRaises(ValueError) as cm:
|
||||
_ = AutoModel.from_pretrained("hf-tiny-model-private/tiny-random-MCTCTModel", attn_implementation="sdpa")
|
||||
|
||||
Reference in New Issue
Block a user