diff --git a/tests/test_lazy/test_models.py b/tests/test_lazy/test_models.py index 0a919955f..fa4075c1e 100644 --- a/tests/test_lazy/test_models.py +++ b/tests/test_lazy/test_models.py @@ -28,10 +28,10 @@ def test_models_lazy_init(subset, default_device): "timm_deit3", "timm_convit", "timm_tnt_b_patch16_224", - ) or name.startswith(("transformers_vit", "transformers_blip2", "transformers_whisper")): + ) or name.startswith(("transformers_vit", "transformers_blip2", "transformers_whisper", "transformers_deepseek")): continue check_lazy_init(entry, verbose=True, default_device=default_device) if __name__ == "__main__": - test_models_lazy_init("transformers", "cpu") + test_models_lazy_init("transformers", "cpu") \ No newline at end of file