diff --git a/llm-apps-course/Generation.ipynb b/llm-apps-course/Generation.ipynb index 9f42e2e8..140fc27c 100644 --- a/llm-apps-course/Generation.ipynb +++ b/llm-apps-course/Generation.ipynb @@ -127,8 +127,8 @@ "metadata": {}, "outputs": [], "source": [ - "MODEL = \"gpt-3.5-turbo\"\n", - "# MODEL = \"gpt-4\"" + "MODEL_NAME = \"gpt-3.5-turbo\"\n", + "# MODEL_NAME = \"gpt-4\"" ] }, { @@ -146,7 +146,7 @@ " {\"role\": \"user\", \"content\": user_prompt},\n", " ]\n", " responses = completion_with_backoff(\n", - " model=MODEL,\n", + " model=MODEL_NAME,\n", " messages=messages,\n", " n = n,\n", " )\n", @@ -273,7 +273,7 @@ "metadata": {}, "outputs": [], "source": [ - "tokenizer = tiktoken.encoding_for_model(MODEL)\n", + "tokenizer = tiktoken.encoding_for_model(MODEL_NAME)\n", "tokens_per_document = [len(tokenizer.encode(document)) for _, document in documents]\n", "pprint(tokens_per_document)" ]