From 58e33776fdb19b76933942b5967cc883742bb645 Mon Sep 17 00:00:00 2001 From: wangbluo <2538539015@qq.com> Date: Mon, 2 Jun 2025 11:21:40 +0800 Subject: [PATCH] fix_lazy_init_deepseek --- tests/test_lazy/test_models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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