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 empty device issue (#904)

Browse Source
This commit is contained in:
Sebastian Raschka
2025-11-05 20:04:44 -06:00
committed by GitHub
parent bcc73f731d
commit 58f45ae5a7
6 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1156,6 +1156,8 @@
" major, minor = map(int, torch.__version__.split(\".\")[:2])\n",
" if (major, minor) >= (2, 9):\n",
" device = torch.device(\"mps\")\n",
" else:\n",
" device = torch.device(\"cpu\")\n",
"else:\n",
" device = torch.device(\"cpu\")\n",
"\n",