Main author: black_samorez@

Co-authored-by: Andrei Panferov <[email protected]>
yandexdataschool · Nov 23, 2023 · b4a5fae · b4a5fae
1 parent 9a0a8cc
commit b4a5fae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/week10_efficiency/benchmark.ipynb b/week10_efficiency/benchmark.ipynb
@@ -50,7 +50,7 @@
         "\n",
         "import torch\n",
         "from torch import Tensor\n",
-        "from torch import nn\n",
+        "import torch.nn as nn\n",
         "\n",
         "from transformers.models.llama.modeling_llama import LlamaForCausalLM\n",
         "from transformers.models.llama.configuration_llama import LlamaConfig\n",