Handle case where lock is left over from failed partial download

This commit is contained in:
Nicolas Mowen 2024-10-12 11:28:42 -06:00
parent f58a2b8eba
commit 8b3bb1bad2
2 changed files with 7 additions and 1 deletions

View File

@ -71,7 +71,6 @@ class GenericONNXEmbedding:
model_name=self.model_name,
download_path=self.download_path,
file_names=files_names,
requestor=self.requestor,
download_func=self._download_model,
)
self.downloader.ensure_model_files()

View File

@ -19,6 +19,13 @@ class FileLock:
self.path = path
self.lock_file = f"{path}.lock"
# we have not acquired the lock yet so it should not exist
if (os.path.exists(self.lock_file)):
try:
os.remove(self.lock_file)
except Exception:
pass
def acquire(self):
parent_dir = os.path.dirname(self.lock_file)
os.makedirs(parent_dir, exist_ok=True)