Merge pull request #53 from gizatechxyz/fix-output_dtype-in-giza-model

Fix output dtype in giza model
gizatechxyz · May 28, 2024 · 167638c · 167638c
2 parents 31d91f6 + cecc42b
commit 167638c
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: