tacit/nanochat
1
0
Fork 0
mirror of https://github.com/karpathy/nanochat.git synced 2026-03-21 20:33:23 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix MockModel's device definition (#535)

Browse Source 
* fix MockModel's device definition

* cleanup
This commit is contained in:
Sofie Van Landeghem 2026-02-18 01:03:46 +01:00 committed by Andrej Karpathy
parent f5fe7925ed
commit cac43e8511
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_engine.py
View File

@ -31,7 +31,7 @@ class MockModel:
def __init__(self, vocab_size=262): # 256 bytes + 6 special tokens
self.vocab_size = vocab_size
self.config = MockConfig()
self._device = "cpu"
self._device = torch.device("cpu")
def get_device(self):
return self._device