KevinMidboe/immich
1
0
Fork 0
mirror of https://github.com/KevinMidboe/immich.git synced 2025-10-29 17:40:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(ml): clear cache if bad zip file (#3600)

Browse Source
This commit is contained in:
Mert
2023-08-08 13:33:13 -04:00
committed by GitHub
parent ddd4ec2d9e
commit fa57853bd2
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
machine-learning/app/models/base.py
View File

@@ -4,6 +4,7 @@ from abc import ABC, abstractmethod
from pathlib import Path from pathlib import Path
from shutil import rmtree from shutil import rmtree
from typing import Any from typing import Any
from zipfile import BadZipFile
from onnxruntime.capi.onnxruntime_pybind11_state import InvalidProtobuf # type: ignore from onnxruntime.capi.onnxruntime_pybind11_state import InvalidProtobuf # type: ignore
@@ -23,7 +24,7 @@ class InferenceModel(ABC):
loader = self.load if eager else self.download loader = self.load if eager else self.download
try: try:
loader(**model_kwargs) loader(**model_kwargs)
except (OSError, InvalidProtobuf): except (OSError, InvalidProtobuf, BadZipFile):
self.clear_cache() self.clear_cache()
loader(**model_kwargs) loader(**model_kwargs)