diff --git a/utils/inference.py b/utils/inference.py index d77dc25..89b30a0 100644 --- 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