Update transformer.py

transformer.py

@@ -5,7 +5,7 @@ import torch.nn as nn
 from torch.nn import functional as F
 import math
 
-from constants import n_head, n_embd, n_layer, n_hidden, feed_forward, n_output
+from constants import n_head, n_embd, n_layer, n_hidden, feed_forward, n_output, block_size
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")