Update src/deepsparse/transformers/infer.py

neuralmagic · Oct 7, 2023 · 4bf02ff · 4bf02ff
1 parent 15f4698
commit 4bf02ff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/deepsparse/transformers/infer.py b/src/deepsparse/transformers/infer.py
@@ -181,10 +181,10 @@ def _display_generation_speed(prompt_sequence_length, pipeline):
 def _display_bot_response(stream: bool, response):
     # print response from pipeline, streaming or not
 
-    print("Bot:", end=" ")
+    print("Bot:", end="")
     if stream:
         for generation in response:
-            print(generation.generations[0].text, end=" ")
+            print(generation.generations[0].text, end="")
         print()
     else:
         print(response.generations[0].text)