github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[test] fix qwen2 pytest distLarge (#5797)

pull/5781/head
Guangyao Zhang 2024-06-12 12:13:51 +08:00 committed by GitHub
parent aa125bcc91
commit aac941ef78
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show Stats 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():