diff --git a/tests/lightning/test_dist_ckpt.py b/tests/lightning/test_dist_ckpt.py index d5037f0aa5731..886b1085ed55e 100644 --- a/tests/lightning/test_dist_ckpt.py +++ b/tests/lightning/test_dist_ckpt.py @@ -34,6 +34,7 @@ def set_env(): def _get_strategy(): strategy = nl.MegatronStrategy( enable_nemo_ckpt_io=False, + ckpt_async_save=False, ) return strategy