From 6f19618bb4ac293fabdbee0aececa918c97aa9c7 Mon Sep 17 00:00:00 2001 From: Hanks Date: Mon, 2 Jun 2025 11:31:45 +0800 Subject: [PATCH] [fix] fix_lazy_init for deepseek model in transformers --- tests/test_lazy/test_models.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_lazy/test_models.py b/tests/test_lazy/test_models.py index 0a919955f..f65952e43 100644 --- a/tests/test_lazy/test_models.py +++ b/tests/test_lazy/test_models.py @@ -28,7 +28,9 @@ 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)