Skip to content

Commit c380f0c

Browse files
committed
misc: add "files deleted" warning in doc string
1 parent b7906db commit c380f0c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

checkpoint_engine/ps.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -969,6 +969,8 @@ def register_checkpoint(
969969
) -> None:
970970
"""
971971
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.
972974
973975
Args:
974976
checkpoint_name: The name of the checkpoint.

0 commit comments

Comments
 (0)