From 9f3982753d9cf3c16073fa5f93efd7f56fec6eb1 Mon Sep 17 00:00:00 2001 From: llermaly Date: Wed, 9 Oct 2024 21:12:49 -0500 Subject: [PATCH] second attempt --- .../using-openelm-models/using-openelm-models.ipynb | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/supporting-blog-content/using-openelm-models/using-openelm-models.ipynb b/supporting-blog-content/using-openelm-models/using-openelm-models.ipynb index fe8dfb6d..4bab37c7 100644 --- a/supporting-blog-content/using-openelm-models/using-openelm-models.ipynb +++ b/supporting-blog-content/using-openelm-models/using-openelm-models.ipynb @@ -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",