fix typo in nanogpt python code

feenkcom · Aug 27, 2024 · 948c61f · 948c61f
1 parent f239d76
commit 948c61f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/nanogpt/__init__.py b/python/nanogpt/__init__.py
@@ -171,7 +171,7 @@ def generate(self, idx, max_new_tokens):
 
 @dataclass
 class TrainingContext:
-	model = nn.Model
+	model = nn.Module
 	context = torch.Tensor
 
 	@gtView