xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-04-30 02:16:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Bugfix] use SoftLockFile instead of LockFile (#3578)

Browse Source
This commit is contained in:
kota-iizuka 2024-03-24 03:43:11 +09:00 committed by GitHub
parent bfdb1ba5c3
commit 743a0b7402
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/model_executor/weight_utils.py
View File

@ -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