books/LLMs-from-scratch
1
0
Fork 0
mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2026-04-10 12:33:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix misplaced parenthesis and update license (#466)

Browse Source
This commit is contained in:
Sebastian Raschka
2025-01-04 11:14:08 -06:00
committed by GitHub
parent 57fdd94358
commit 05a816e270
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ch05/01_main-chapter-code/gpt_train.py
View File

@@ -239,4 +239,4 @@ if __name__ == "__main__":
# Save and load model
torch.save(model.state_dict(), "model.pth")
model = GPTModel(GPT_CONFIG_124M)
model.load_state_dict(torch.load("model.pth"), weights_only=True)
model.load_state_dict(torch.load("model.pth", weights_only=True))