diff --git a/tests/a_scripts/inference/test_convert_composer_to_hf.py b/tests/a_scripts/inference/test_convert_composer_to_hf.py index d21c942dee..de85be0d0c 100644 --- a/tests/a_scripts/inference/test_convert_composer_to_hf.py +++ b/tests/a_scripts/inference/test_convert_composer_to_hf.py @@ -391,7 +391,7 @@ def test_huggingface_conversion_callback(model: str, tmp_path: pathlib.Path, expected_normal_checkpoints: int): delete_transformers_cache() - dist.initialize_dist(get_device('gpu')) + # dist.initialize_dist(get_device('gpu')) max_seq_len = 16 device_batch_size = 1