diff --git a/model.py b/model.py index 93d0120a..08bc23f9 100644 --- a/model.py +++ b/model.py @@ -705,7 +705,7 @@ def generate_gen(self, prompt: str, **kwargs): extra_parts.append("<-- Not accurate (truncated)") # Print output - print( + logger.info( initial_response + " (" + ", ".join(itemization)