aparler/c-llm
1
0
Fork 0
mirror of https://github.com/karpathy/llm.c.git synced 2026-07-28 20:35:09 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: torch warning of python demo

Browse source
This commit is contained in:
DominguesAddem1974 2024-04-09 18:17:04 +08:00
parent a22c22b1ee
commit daf1dd81b2
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
train_gpt2.py
View file

@ -330,7 +330,7 @@ if __name__ == "__main__":
tokens = np.frombuffer(f.read(), dtype=np.int32)
# np -> tensor, long, on device
tokens = torch.from_numpy(tokens)
tokens = torch.tensor(tokens)
tokens = tokens.to(torch.long)
tokens = tokens.to(device)