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

fix MockModel's device definition

Browse Source
This commit is contained in:
svlandeg 2026-02-15 20:30:35 +01:00
parent 2f09686724
commit 07f1a3e8cf
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_engine.py
View File

@ -32,6 +32,7 @@ class MockModel:
self.vocab_size = vocab_size
self.config = MockConfig()
self._device = "cpu"
self._device = torch.device("cpu")
def get_device(self):
return self._device