Skip to content

Commit 6f19618

Browse files
authored
[fix] fix_lazy_init for deepseek model in transformers
1 parent d322ff8 commit 6f19618

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/test_lazy/test_models.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,9 @@ def test_models_lazy_init(subset, default_device):
2828
"timm_deit3",
2929
"timm_convit",
3030
"timm_tnt_b_patch16_224",
31-
) or name.startswith(("transformers_vit", "transformers_blip2", "transformers_whisper")):
31+
) or name.startswith(
32+
("transformers_vit", "transformers_blip2", "transformers_whisper", "transformers_deepseek")
33+
):
3234
continue
3335
check_lazy_init(entry, verbose=True, default_device=default_device)
3436

0 commit comments

Comments
 (0)