naming consistency

sugatoray · May 31, 2023 · 4b9908a · 4b9908a
1 parent 2228f0c
commit 4b9908a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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)"
    ]