From b143433264b1d894a2008478fba110bbd11480e7 Mon Sep 17 00:00:00 2001 From: raphaelDkhn Date: Tue, 28 May 2024 11:10:25 +0300 Subject: [PATCH 1/2] Update model.py --- giza/agents/model.py | 1 + 1 file changed, 1 insertion(+) diff --git a/giza/agents/model.py b/giza/agents/model.py index db14076..bd7359d 100644 --- 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 = "default_dtype" # Define a default or determine a better initial value based on context try: logger.info("Predicting") if verifiable: From cecc42b890203c3dbd6e62d8bf5389ff9ebdc78b Mon Sep 17 00:00:00 2001 From: raphaelDkhn Date: Tue, 28 May 2024 11:11:45 +0300 Subject: [PATCH 2/2] define default output_dtype --- giza/agents/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/giza/agents/model.py b/giza/agents/model.py index bd7359d..8ef9209 100644 --- 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" try: logger.info("Predicting") if verifiable: