From c5e023fbe64641bc1d7ea257bc58fa030137c60a Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Thu, 15 Jan 2026 13:02:55 +0530 Subject: [PATCH] up --- tests/models/test_models_auto.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/tests/models/test_models_auto.py b/tests/models/test_models_auto.py index a70754343f..c0e9b4494b 100644 --- a/tests/models/test_models_auto.py +++ b/tests/models/test_models_auto.py @@ -20,7 +20,9 @@ class TestAutoModel(unittest.TestCase): side_effect=[EnvironmentError("File not found"), {"model_type": "clip_text_model"}], ) def test_load_from_config_transformers_with_subfolder(self, mock_load_config): - model = AutoModel.from_pretrained("hf-internal-testing/tiny-stable-diffusion-torch", subfolder="text_encoder") + model = AutoModel.from_pretrained( + "hf-internal-testing/tiny-stable-diffusion-torch", subfolder="text_encoder", use_safetensors=False + ) assert isinstance(model, CLIPTextModel) def test_load_from_config_without_subfolder(self): @@ -28,5 +30,7 @@ class TestAutoModel(unittest.TestCase): assert isinstance(model, LongformerModel) def test_load_from_model_index(self): - model = AutoModel.from_pretrained("hf-internal-testing/tiny-stable-diffusion-torch", subfolder="text_encoder") + model = AutoModel.from_pretrained( + "hf-internal-testing/tiny-stable-diffusion-torch", subfolder="text_encoder", use_safetensors=False + ) assert isinstance(model, CLIPTextModel)