diff --git a/chapter12_part01_text-generation.ipynb b/chapter12_part01_text-generation.ipynb index f683c1d73b..b84f43c696 100644 --- a/chapter12_part01_text-generation.ipynb +++ b/chapter12_part01_text-generation.ipynb @@ -407,6 +407,7 @@ " predictions = self.model(tokenized_sentence)\n", " next_token = sample_next(\n", " predictions[0, self.prompt_length - 1 + i, :]\n", + " temperature " )\n", " sampled_token = tokens_index[next_token]\n", " sentence += \" \" + sampled_token\n", @@ -478,4 +479,4 @@ }, "nbformat": 4, "nbformat_minor": 0 -} \ No newline at end of file +}