From c4ffc20a1afcd3897b2e5e86824c20c2a66630a0 Mon Sep 17 00:00:00 2001 From: kota-iizuka <64062831+kota-iizuka@users.noreply.github.com> Date: Sat, 23 Mar 2024 06:19:17 +0000 Subject: [PATCH] use SoftLockFile instead of LockFile --- vllm/model_executor/weight_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/model_executor/weight_utils.py b/vllm/model_executor/weight_utils.py index 24d78db8d263..0d7ee269922f 100644 --- a/vllm/model_executor/weight_utils.py +++ b/vllm/model_executor/weight_utils.py @@ -34,7 +34,7 @@ def get_lock(model_name_or_path: str, cache_dir: Optional[str] = None): lock_dir = cache_dir if cache_dir is not None else _vllm_filelocks_path os.makedirs(os.path.dirname(lock_dir), exist_ok=True) lock_file_name = model_name_or_path.replace("/", "-") + ".lock" - lock = filelock.FileLock(os.path.join(lock_dir, lock_file_name)) + lock = filelock.SoftFileLock(os.path.join(lock_dir, lock_file_name)) return lock