Fix FSDP ckpt by filtering User Waring (mosaicml#3327)

mvpatel2000 · May 25, 2024 · 0be5d30 · 0be5d30
1 parent 63b9473
commit 0be5d30
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/trainer/test_fsdp_checkpoint.py b/tests/trainer/test_fsdp_checkpoint.py
@@ -481,6 +481,7 @@ def test_fsdp_mixed_with_sync(
 )
 @pytest.mark.filterwarnings(r'ignore:.*metrics are not saved with sharded state dict.*:UserWarning')
 @pytest.mark.filterwarnings(r'ignore:.*The CUDA RNG state could not be loaded.*:UserWarning')
+@pytest.mark.filterwarnings(r'ignore:.*ShardedTensor.to only move tensor to its current device.*:UserWarning')
 def test_fsdp_load_old_checkpoint(
     world_size,
     tmp_path: pathlib.Path,