github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-24 20:20:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

[fix] fix_lazy_init for deepseek model in transformers

Browse Source
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
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_lazy/test_models.py
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)