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

Merge a991ed8e39 into 8180e1d8c1

Browse Source
This commit is contained in:
Sofie Van Landeghem 2026-02-16 23:12:51 +01:00 committed by GitHub
commit 969e8980e7
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