git-market/rasbt-LLMs-from-scratch
1
0
Fork 0
mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2026-04-11 02:11:40 +08:00
Code Issues Actions 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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
6 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -93,6 +93,8 @@
" device = torch.device(\"mps\")\n",
" else:\n",
" device = torch.device(\"cpu\")\n",
"else:\n",
" device = torch.device(\"cpu\")\n",
"\n",
"print(\"Device:\", device)\n",
"\n",

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

@ -563,6 +563,8 @@
" device = torch.device(\"mps\")\n",
" else:\n",
" device = torch.device(\"cpu\")\n",
"else:\n",
" device = torch.device(\"cpu\")\n",
"\n",
"print(\"Device:\", device)\n",
"\n",

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

@ -1185,6 +1185,9 @@
" device = torch.device(\"mps\")\n",
" else:\n",
" device = torch.device(\"cpu\")\n",
"else:\n",
" device = torch.device(\"cpu\")\n",
"\n",
"\n",
"print(f\"Using {device} device.\")\n",
"\n",

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

@ -1897,6 +1897,8 @@
" device = torch.device(\"mps\")\n",
" else:\n",
" device = torch.device(\"cpu\")\n",
"else:\n",
" device = torch.device(\"cpu\")\n",
"\n",
"print(\"Device:\", device)\n",
"\n",

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

@ -1158,6 +1158,8 @@
" device = torch.device(\"mps\")\n",
" else:\n",
" device = torch.device(\"cpu\")\n",
"else:\n",
" device = torch.device(\"cpu\")\n",
"\n",
"print(\"Device:\", device)"
]

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

@ -727,6 +727,8 @@
" device = torch.device(\"mps\")\n",
" else:\n",
" device = torch.device(\"cpu\")\n",
"else:\n",
" device = torch.device(\"cpu\")\n",
"\n",
"print(\"Device:\", device)\n",
"\n",