missed one test

Signed-off-by: Peter St. John <pstjohn@nvidia.com>

Author: pstjohn

recipes/geneformer_native_te_nvfsdp_fp8/test_distributed_checkpointing.py

@@ -797,6 +797,7 @@ def test_safetensors_multiprocess_roundtrip_nvfsdp():
 
 
 @pytest.mark.slow
+@requires_multi_gpu
 def test_safetensors_unsharded_weights_consistency():
     """Test that unsharded weights from multiprocess training match single-process training.