diff --git a/tests/e2e/multigpu/test_llama.py b/tests/e2e/multigpu/test_llama.py index c2bb12a02..f00c54bad 100644 --- a/tests/e2e/multigpu/test_llama.py +++ b/tests/e2e/multigpu/test_llama.py @@ -894,7 +894,7 @@ class TestMultiGPULlama: "sample_packing": True, "bf16": True, "save_safetensors": True, - "deepspeed": str(AXOLOTL_ROOT / "deepspeed_configs/zero1.json"), + # "deepspeed": str(AXOLOTL_ROOT / "deepspeed_configs/zero1.json"), "use_tensorboard": True, } )