Skip to content

Commit

Permalink
Remove debug prompt print (#372)
Browse files Browse the repository at this point in the history
  • Loading branch information
natke authored May 2, 2024
1 parent d7d82a3 commit 2413e3b
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions examples/python/phi3-qa.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,8 +29,6 @@ def main(args):
# If there is a chat template, use it
prompt = f'{chat_template.format(input=text)}'

print(f"Prompt: {prompt}")

input_tokens = tokenizer.encode(prompt)

params = og.GeneratorParams(model)
Expand Down Expand Up @@ -84,4 +82,4 @@ def main(args):
parser.add_argument('-v', '--verbose', action='store_true', default=False, help='Print verbose output and timing information. Defaults to false')
parser.add_argument('-g', '--timings', action='store_true', default=False, help='Print timing information for each generation step. Defaults to false')
args = parser.parse_args()
main(args)
main(args)

0 comments on commit 2413e3b

Please sign in to comment.