Skip to content

Commit

Permalink
3rd attempt
Browse files Browse the repository at this point in the history
  • Loading branch information
llermaly committed Oct 10, 2024
1 parent 9f39827 commit de83267
Showing 1 changed file with 1 addition and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -582,8 +582,7 @@
" prompt=prompt,\n",
" model=MODEL,\n",
" hf_access_token=HUGGINGFACE_TOKEN,\n",
" generate_kwargs={\"repetition_penalty\": 1.2,\n",
" \"prompt_lookup_num_tokens\": 10},\n",
" generate_kwargs={\"repetition_penalty\": 1.2, \"prompt_lookup_num_tokens\": 10},\n",
")\n",
"print(\"-----GENERATION TIME-----\")\n",
"print(f\"\\033[92m {round(generation_time, 2)} \\033[0m\")\n",
Expand Down

0 comments on commit de83267

Please sign in to comment.