Skip to content

Commit

Permalink
formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
rishsriv committed Jan 10, 2024
1 parent c9ca53b commit 4b60649
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion eval/hf_runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -113,7 +113,7 @@ def run_hf_eval(args):
pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)

support_beam_search = True

# check that the model supports beam search with a try except statement
try:
pipe("Hi", num_beams=2, do_sample=False, max_new_tokens=5)
Expand Down

0 comments on commit 4b60649

Please sign in to comment.