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

release

Browse Source
This commit is contained in:
flybird11111 2025-05-27 14:38:59 +08:00
parent ddbbbaab3e
commit 4271e3daf6
3 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/kit/model_zoo/transformers/gpt.py
View File

@ -113,6 +113,7 @@ config = transformers.GPT2Config(
problem_type="single_label_classification", problem_type="single_label_classification",
pad_token_id=1022, pad_token_id=1022,
tie_word_embeddings=True, tie_word_embeddings=True,
attn_implementation="flash_attention_2",
) )
config_for_token_classification = copy.deepcopy(config) config_for_token_classification = copy.deepcopy(config)

1
tests/test_zero/test_gemini/test_inference.py
View File

@ -114,7 +114,6 @@ def run_dist(rank, world_size, port):
exam_inference() exam_inference()
@pytest.mark.skip("this test failed")
@pytest.mark.dist @pytest.mark.dist
@pytest.mark.parametrize("world_size", [1, 4]) @pytest.mark.parametrize("world_size", [1, 4])
def test_inference(world_size): def test_inference(world_size):

2
version.txt
View File

@ -1 +1 @@
0.4.9 0.5.0