Skip to content

Commit 757d1aa

Browse files
authored
Pass delete from disk to clear cache (#1982)
* Pass delete from disk to clear cache
1 parent 074965e commit 757d1aa

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

inference/core/models/roboflow.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -934,7 +934,7 @@ def initialize_model(self, **kwargs) -> None:
934934
sess_options=session_options,
935935
)
936936
except Exception as e:
937-
self.clear_cache()
937+
self.clear_cache(delete_from_disk=DISK_CACHE_CLEANUP)
938938
raise ModelArtefactError(
939939
f"Unable to load ONNX session. Cause: {e}"
940940
) from e

inference/models/rfdetr/rfdetr.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
from inference.core.entities.responses.inference import InferenceResponseImage
1313
from inference.core.env import (
1414
DISABLE_PREPROC_AUTO_ORIENT,
15+
DISK_CACHE_CLEANUP,
1516
FIX_BATCH_SIZE,
1617
MAX_BATCH_SIZE,
1718
ONNXRUNTIME_EXECUTION_PROVIDERS,
@@ -471,7 +472,7 @@ def initialize_model(self, **kwargs) -> None:
471472
sess_options=session_options,
472473
)
473474
except Exception as e:
474-
self.clear_cache()
475+
self.clear_cache(delete_from_disk=DISK_CACHE_CLEANUP)
475476
raise ModelArtefactError(
476477
f"Unable to load ONNX session. Cause: {e}"
477478
) from e

0 commit comments

Comments
 (0)