diff --git a/tests/utils/test_modeling_utils.py b/tests/utils/test_modeling_utils.py index cae38422f532..970dc5994041 100644 --- a/tests/utils/test_modeling_utils.py +++ b/tests/utils/test_modeling_utils.py @@ -2608,7 +2608,8 @@ def test_not_available_flash(self): _ = AutoModel.from_pretrained( "hf-internal-testing/tiny-random-GPTBigCodeModel", attn_implementation="flash_attention_2" ) - self.assertTrue("the package flash_attn seems to be not installed" in str(cm.exception)) + + self.assertTrue("the package flash_attn seems not to be installed" in str(cm.exception)) def test_not_available_flash_with_config(self): if is_flash_attn_2_available(): @@ -2623,7 +2624,7 @@ def test_not_available_flash_with_config(self): attn_implementation="flash_attention_2", ) - self.assertTrue("the package flash_attn seems to be not installed" in str(cm.exception)) + self.assertTrue("the package flash_attn seems not to be installed" in str(cm.exception)) def test_not_available_sdpa(self): if is_torch_sdpa_available():