File tree Expand file tree Collapse file tree 2 files changed +3
-2
lines changed
Expand file tree Collapse file tree 2 files changed +3
-2
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff line change 1212from inference .core .entities .responses .inference import InferenceResponseImage
1313from 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
You can’t perform that action at this time.
0 commit comments