[fix] fix_lazy_init for deepseek model in transformers

This commit is contained in:
Hanks 2025-06-02 11:31:45 +08:00 committed by GitHub
parent d322ff8cd9
commit 6f19618bb4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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)