github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-21 05:04:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #6335 from wangbluo/lazy_deepseek

Browse Source 
[fix]fix_lazy_init for deepseek model in transformers
This commit is contained in:
Hanks 2025-06-02 11:30:11 +08:00 committed by GitHub
commit 5890c8ecdd
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)