diff --git a/tests/test_thunder_fsdp.py b/tests/test_thunder_fsdp.py index 8b9c0f4340..321cdac7a6 100644 --- a/tests/test_thunder_fsdp.py +++ b/tests/test_thunder_fsdp.py @@ -263,6 +263,8 @@ def set_up_planner(self, state_dict, metadata, is_coordinator): @RunIf(min_cuda_gpus=2, thunder=True, standalone=True) def test_save_load_sharded_checkpoint(tmp_path): + pytest.skip("Temporarily disabled, often exceeds 5 min timeout") + strategy = ThunderFSDPStrategy(state_dict_type="sharded", broadcast_from=0) fabric = Fabric(accelerator="cuda", devices=2, strategy=strategy) fabric.launch()