Skip to content

Commit

Permalink
skip conversion test due to multiprocessing issue
Browse files Browse the repository at this point in the history
  • Loading branch information
AI-WAIFU committed Oct 8, 2024
1 parent 44a2077 commit 3bb9128
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion tests/unit/test_format_conversion_scripts.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,9 +3,12 @@
from tests.common import simulate_deepy_env, save_random_model
from megatron.neox_arguments.neox_args import NeoXArgsTokenizer


@pytest.mark.skip(
reason="Conversion test is skipped until we fix the CUDA + torch multiprocessing issue."
)
def test_gpt_neox_to_huggingface(monkeypatch, tmpdir, tmp_path):
# Generate random GPT-NEOX model, check we can convert to hf format

model_dir = str(tmpdir)
input_args = ["train.py", "tests/config/test_setup.yml"]
deepspeed_main_args = simulate_deepy_env(monkeypatch, input_args)
Expand Down

0 comments on commit 3bb9128

Please sign in to comment.