From b4a5faed6062ad09a612c56d357ea4439facee87 Mon Sep 17 00:00:00 2001
From: Your Name <justheuristic@gmail.com>
Date: Thu, 23 Nov 2023 20:21:04 +0300
Subject: [PATCH] Main author: black_samorez@

Co-authored-by: Andrei Panferov <andrei@BlackSamorez.ru>
---
 week10_efficiency/benchmark.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/week10_efficiency/benchmark.ipynb b/week10_efficiency/benchmark.ipynb
index f861861c..a614d505 100644
--- 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",