Skip to content

Commit

Permalink
second attempt
Browse files Browse the repository at this point in the history
  • Loading branch information
llermaly committed Oct 10, 2024
1 parent ee4bb26 commit 9f39827
Showing 1 changed file with 2 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -582,7 +582,8 @@
" prompt=prompt,\n",
" model=MODEL,\n",
" hf_access_token=HUGGINGFACE_TOKEN,\n",
" generate_kwargs={\"repetition_penalty\": 1.2, \"prompt_lookup_num_tokens\": 10},\n",
" generate_kwargs={\"repetition_penalty\": 1.2,\n",
" \"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 9f39827

Please sign in to comment.