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

Merge a991ed8e39 into 788dadeb88

Browse Source
This commit is contained in:
Sofie Van Landeghem 2026-02-16 14:50:44 +00:00 committed by GitHub
commit e093326f0e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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