We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b7906db commit c380f0cCopy full SHA for c380f0c
checkpoint_engine/ps.py
@@ -969,6 +969,8 @@ def register_checkpoint(
969
) -> None:
970
"""
971
Register a checkpoint to the parameter server. Both files and named_tensors will be registered together.
972
+ Warning: .safetensors files in /dev/shm/ will be pinned in-place, and the files will be REMOVED after pinning.
973
+ Please make sure to copy the files to disks if you need to keep them.
974
975
Args:
976
checkpoint_name: The name of the checkpoint.
0 commit comments