fam/nanochat-omni
1
0
Fork 0
Code Issues Pull Requests 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 GitHub
parent 4a6e47b0c6
commit 4800c62f6e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_engine.py
+1 -1
View File
@@ -31,7 +31,7 @@ class MockModel:
def __init__(self, vocab_size=262): # 256 bytes + 6 special tokens def __init__(self, vocab_size=262): # 256 bytes + 6 special tokens
self.vocab_size = vocab_size self.vocab_size = vocab_size
self.config = MockConfig() self.config = MockConfig()
self._device = "cpu" self._device = torch.device("cpu")
def get_device(self): def get_device(self):
return self._device return self._device