feat: rename variables

dev/inference/wandb-backend.ipynb

@@ -50,7 +50,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "run_ids = ['he9rrc3q']\n",
+    "run_ids = ['2u5lk3uw']\n",
     "# poorly shuffled 1nj161cl\n",
     "# well shuffled he9rrc3q\n",
     "# non normalized 1fwxpyfh ! requires changing normalize_text\n",
@@ -70,8 +70,8 @@
    "outputs": [],
    "source": [
     "# model.generate parameters - Not used yet\n",
-    "top_k = None\n",
-    "top_p = None\n",
+    "gen_top_k = None\n",
+    "gen_top_p = None\n",
     "temperature = None"
    ]
   },
@@ -213,7 +213,9 @@
     "            do_sample=True,\n",
     "            num_beams=1,\n",
     "            prng_key=key,\n",
-    "            params=params\n",
+    "            params=params,\n",
+    "            top_k=gen_top_k,\n",
+    "            top_p=gen_top_p\n",
     "        )\n",
     "    \n",
     "    return _generate"