Update utils.py

utils.py

@@ -11,6 +11,7 @@ n_embd = 128
 n_head = 4
 n_layer = 8
 dropout = 0.1
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
 class Head(nn.Module):
     """ one head of self-attention """