diff --git a/tests/models/auto/test_modeling_tf_auto.py b/tests/models/auto/test_modeling_tf_auto.py index 2b4b625e2305..1a355d88bb5a 100644 --- a/tests/models/auto/test_modeling_tf_auto.py +++ b/tests/models/auto/test_modeling_tf_auto.py @@ -281,7 +281,7 @@ def test_revision_not_found(self): def test_model_file_not_found(self): with self.assertRaisesRegex( EnvironmentError, - "hf-internal-testing/config-no-model does not appear to have a file named tf_model.h5", + "hf-internal-testing/config-no-model does not appear to have a file named pytorch_model.bin", ): _ = TFAutoModel.from_pretrained("hf-internal-testing/config-no-model")