comment a print

Tlntin · Jul 31, 2024 · 4429193 · 4429193
1 parent e4c350f
commit 4429193
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/inference.py b/utils/inference.py
@@ -145,7 +145,7 @@ def stream_predict(
             [text], return_tensors="np"
         )["input_ids"].astype(np.int64).reshape(1, -1)
         input_ids = input_ids[:, -self.max_input_length:]
-        print("input_ids shape: ", input_ids.shape)
+        # print("input_ids shape: ", input_ids.shape)
         self.first = False
         ids_list = []
         text_length = 0