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
appendix-D/01_main-chapter-code/appendix-D.ipynb
View File

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

2
appendix-E/01_main-chapter-code/appendix-E.ipynb
View File

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

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

@@ -1183,9 +1183,12 @@
" major, minor = map(int, torch.__version__.split(\".\")[:2])\n", " major, minor = map(int, torch.__version__.split(\".\")[:2])\n",
" if (major, minor) >= (2, 9):\n", " if (major, minor) >= (2, 9):\n",
" device = torch.device(\"mps\")\n", " device = torch.device(\"mps\")\n",
" else:\n",
" device = torch.device(\"cpu\")\n",
"else:\n", "else:\n",
" device = torch.device(\"cpu\")\n", " device = torch.device(\"cpu\")\n",
"\n", "\n",
"\n",
"print(f\"Using {device} device.\")\n", "print(f\"Using {device} device.\")\n",
"\n", "\n",
"\n", "\n",

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

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

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

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

2
ch07/04_preference-tuning-with-dpo/dpo-from-scratch.ipynb
View File

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