Upload instruct_pipeline.py

instruct_pipeline.py

@@ -105,7 +105,6 @@ class InstructionTextGenerationPipeline(Pipeline):
             **generate_kwargs,
         )[0].cpu()
         instruction_text = model_inputs.pop("instruction_text")
-        print(instruction_text)
         return {"generated_sequence": generated_sequence, "input_ids": input_ids, "instruction_text": instruction_text}
 
     def postprocess(self, model_outputs, response_key_token_id, end_key_token_id, return_instruction_text):
@@ -142,8 +141,6 @@ class InstructionTextGenerationPipeline(Pipeline):
 
             fully_decoded = self.tokenizer.decode(sequence)
 
-            print(fully_decoded)
-
             # The response appears after "### Response:".  The model has been trained to append "### End" at the
             # end.
             m = re.search(r"#+\s*Response:\s*(.+?)#+\s*End", fully_decoded, flags=re.DOTALL)