mirror of
https://github.com/rasbt/LLMs-from-scratch.git
synced 2026-04-10 12:33:42 +00:00
Fix empty device issue (#904)
This commit is contained in:
committed by
GitHub
parent
bcc73f731d
commit
58f45ae5a7
@@ -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",
|
||||
|
||||
@@ -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",
|
||||
|
||||
@@ -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",
|
||||
|
||||
@@ -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",
|
||||
|
||||
@@ -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)"
|
||||
]
|
||||
|
||||
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user