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

fix_lazy_init_deepseek

Browse Source
This commit is contained in:
wangbluo 2025-06-02 11:21:40 +08:00
parent ba93bba8bf
commit 58e33776fd
1 changed files with 2 additions and 2 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,10 +28,10 @@ def test_models_lazy_init(subset, default_device):
"timm_deit3", "timm_deit3",
"timm_convit", "timm_convit",
"timm_tnt_b_patch16_224", "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 continue
check_lazy_init(entry, verbose=True, default_device=default_device) check_lazy_init(entry, verbose=True, default_device=default_device)
if __name__ == "__main__": if __name__ == "__main__":
test_models_lazy_init("transformers", "cpu") test_models_lazy_init("transformers", "cpu")