From 5727169a2fd41dd919fdf13980c67ef267072455 Mon Sep 17 00:00:00 2001
From: Hooman Shayani <hooshaya@gmail.com>
Date: Mon, 30 Jan 2023 17:49:35 +0000
Subject: [PATCH] removes unnecessary duplicate variable

---
 gpt.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/gpt.py b/gpt.py
index bbde6c05..cad02b69 100644
--- a/gpt.py
+++ b/gpt.py
@@ -184,9 +184,9 @@ def generate(self, idx, max_new_tokens):
         return idx
 
 model = BigramLanguageModel()
-m = model.to(device)
+model.to(device)
 # print the number of parameters in the model
-print(sum(p.numel() for p in m.parameters())/1e6, 'M parameters')
+print(sum(p.numel() for p in model.parameters())/1e6, 'M parameters')
 
 # create a PyTorch optimizer
 optimizer = torch.optim.AdamW(model.parameters(), lr=learning_rate)