diff --git a/giza/agents/model.py b/giza/agents/model.py index fdf98fb..847b697 100644 --- a/giza/agents/model.py +++ b/giza/agents/model.py @@ -184,6 +184,9 @@ def _set_session(self) -> Optional[ort.InferenceSession]: if self._output_path in self._cache: file_path = Path(self._cache.get(self._output_path)) + if file_path.suffix == ".json": + logger.warning("Skipping session for ZKBoost model") + return None with open(file_path, "rb") as f: onnx_model = f.read()