comment out

mosaicml · Dec 1, 2023 · 559f37c · 559f37c
1 parent 0dcaf76
commit 559f37c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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