github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-08 20:40:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

[test] fix qwen2 pytest distLarge (#5797)

Browse Source
This commit is contained in:
Guangyao Zhang
2024-06-12 12:13:51 +08:00
committed by GitHub
parent aa125bcc91
commit aac941ef78
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_shardformer/test_model/test_shard_qwen2.py
View File

@@ -217,6 +217,7 @@ def check_qwen2_3d(rank, world_size, port):
@pytest.mark.skipif(transformers.__version__ < "4.39.1", reason="Requires transformers version 4.39.1 or later")
@pytest.mark.dist
@rerun_if_address_is_in_use()
@clear_cache_before_run()
def test_qwen2():
@@ -224,6 +225,7 @@ def test_qwen2():
@pytest.mark.skipif(transformers.__version__ < "4.39.1", reason="Requires transformers version 4.39.1 or later")
@pytest.mark.largedist
@rerun_if_address_is_in_use()
@clear_cache_before_run()
def test_qwen2_3d():