add comment and support bitwise align

volcengine · Feb 6, 2025 · b9d4307 · b9d4307
1 parent 0a24093
commit b9d4307
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/checkpoint/test_fsdp_ckpt.py b/tests/checkpoint/test_fsdp_ckpt.py
@@ -112,7 +112,7 @@ def test_fsdp_ckpt():
         logits_after_load = model(input_ids=input_ids2, attention_mask=attention_mask2).logits
 
     # Step 4: Verify outputs match
-    torch.testing.assert_close(logits_before_load, logits_after_load)
+    torch.testing.assert_close(logits_before_load, logits_after_load, atol=0.0, rtol=0.0)
     print("Checkpoint save/load test passed!")
 
     # Cleanup

diff --git a/verl/utils/fs.py b/verl/utils/fs.py
@@ -20,7 +20,7 @@
 import hashlib
 
 try:
-    from hdfs_io import copy, makedirs, exists
+    from hdfs_io import copy, makedirs, exists  # for internal use only
 except ImportError:
     from .hdfs_io import copy, makedirs, exists