github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-12 21:25:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse Source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2025-06-02 03:27:11 +00:00
parent 58e33776fd
commit 71f9cedcdb
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_lazy/test_models.py
View File

@ -28,10 +28,12 @@ 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", "transformers_deepseek")): ) 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")