Merge branch 'main' of https://github.com/gizatechxyz/giza-agents

gizatechxyz · May 28, 2024 · 8e8c675 · 8e8c675
2 parents af5ad73 + 167638c
commit 8e8c675
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/giza/agents/model.py b/giza/agents/model.py
@@ -267,6 +267,7 @@ def predict(
         Raises:
             ValueError: If required parameters are not provided or the session is not initialized.
         """
+        output_dtype = "Tensor<FP16x16>"
         try:
             logger.info("Predicting")
             if verifiable: