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

Note about MPS devices (#329)

Browse Source
This commit is contained in:
Sebastian Raschka
2024-08-19 20:58:45 -05:00
committed by GitHub
parent c443035d56
commit 01cb137bfd
3 changed files with 43 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ch05/01_main-chapter-code/ch05.ipynb
View File

@@ -1154,6 +1154,8 @@
"# device = torch.device(\"mps\")\n",
"#else:\n",
"# device = torch.device(\"cpu\")\n",
"#\n",
"# print(f\"Using {device} device.\")\n",
"\n",
"\n",
"model.to(device) # no assignment model = model.to(device) necessary for nn.Module classes\n",