update comments

tomogwen · Feb 29, 2024 · 07a7068 · 07a7068
1 parent 924e05a
commit 07a7068
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/litgpt/model.py b/src/litgpt/model.py
@@ -8,7 +8,6 @@
 from torch.nn import functional as F
 
 # global hyperparams
-# TODO: add these to an args/hparam object
 VOCAB_SIZE = 65
 N_EMBD = 384  # dimension of token embeddings
 N_HEADS = 6  # number of self-attention heads