add diversity

ecs-vlc · Mar 19, 2024 · 82308b4 · 82308b4
1 parent 61e5359
commit 82308b4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/labs/lab7/7_1_SequenceModelling.ipynb b/docs/labs/lab7/7_1_SequenceModelling.ipynb
@@ -1475,7 +1475,7 @@
         "            for i in range(400):\n",
         "                tag_scores = model(inputs)\n",
         "                # 0 because there is a single seq as input, Batch = 1\n",
-        "                c = sample(tag_scores[0])  \n",
+        "                c = sample(tag_scores[0], diversity)  \n",
         "                print(indices_char[c.item()], flush=True, end='')\n",
         "\n",
         "                #adding new generated character to input\n",