mirror of
https://github.com/karpathy/nanochat.git
synced 2026-06-18 20:19:08 +00:00
cleanup
This commit is contained in:
parent
07f1a3e8cf
commit
a991ed8e39
|
|
@ -31,7 +31,6 @@ 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):
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user