If Zkboost dont set the session for now

gizatechxyz · Jun 10, 2024 · 529d644 · 529d644
1 parent 117cfef
commit 529d644
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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()