define default output_dtype

gizatechxyz · May 28, 2024 · cecc42b · cecc42b
1 parent b143433
commit cecc42b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/giza/agents/model.py b/giza/agents/model.py
@@ -267,7 +267,7 @@ def predict(
         Raises:
             ValueError: If required parameters are not provided or the session is not initialized.
         """
-        output_dtype = "default_dtype"  # Define a default or determine a better initial value based on context
+        output_dtype = "Tensor<FP16x16>"
         try:
             logger.info("Predicting")
             if verifiable: