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",
|
" device = torch.device(\"mps\")\n",
|
||||||
" else:\n",
|
" else:\n",
|
||||||
" device = torch.device(\"cpu\")\n",
|
" device = torch.device(\"cpu\")\n",
|
||||||
|
"else:\n",
|
||||||
|
" device = torch.device(\"cpu\")\n",
|
||||||
"\n",
|
"\n",
|
||||||
"print(\"Device:\", device)\n",
|
"print(\"Device:\", device)\n",
|
||||||
"\n",
|
"\n",
|
||||||
|
|||||||
@@ -563,6 +563,8 @@
|
|||||||
" device = torch.device(\"mps\")\n",
|
" device = torch.device(\"mps\")\n",
|
||||||
" else:\n",
|
" else:\n",
|
||||||
" device = torch.device(\"cpu\")\n",
|
" device = torch.device(\"cpu\")\n",
|
||||||
|
"else:\n",
|
||||||
|
" device = torch.device(\"cpu\")\n",
|
||||||
"\n",
|
"\n",
|
||||||
"print(\"Device:\", device)\n",
|
"print(\"Device:\", device)\n",
|
||||||
"\n",
|
"\n",
|
||||||
|
|||||||
@@ -1185,6 +1185,9 @@
|
|||||||
" device = torch.device(\"mps\")\n",
|
" device = torch.device(\"mps\")\n",
|
||||||
" else:\n",
|
" else:\n",
|
||||||
" device = torch.device(\"cpu\")\n",
|
" device = torch.device(\"cpu\")\n",
|
||||||
|
"else:\n",
|
||||||
|
" device = torch.device(\"cpu\")\n",
|
||||||
|
"\n",
|
||||||
"\n",
|
"\n",
|
||||||
"print(f\"Using {device} device.\")\n",
|
"print(f\"Using {device} device.\")\n",
|
||||||
"\n",
|
"\n",
|
||||||
|
|||||||
@@ -1897,6 +1897,8 @@
|
|||||||
" device = torch.device(\"mps\")\n",
|
" device = torch.device(\"mps\")\n",
|
||||||
" else:\n",
|
" else:\n",
|
||||||
" device = torch.device(\"cpu\")\n",
|
" device = torch.device(\"cpu\")\n",
|
||||||
|
"else:\n",
|
||||||
|
" device = torch.device(\"cpu\")\n",
|
||||||
"\n",
|
"\n",
|
||||||
"print(\"Device:\", device)\n",
|
"print(\"Device:\", device)\n",
|
||||||
"\n",
|
"\n",
|
||||||
|
|||||||
@@ -1158,6 +1158,8 @@
|
|||||||
" device = torch.device(\"mps\")\n",
|
" device = torch.device(\"mps\")\n",
|
||||||
" else:\n",
|
" else:\n",
|
||||||
" device = torch.device(\"cpu\")\n",
|
" device = torch.device(\"cpu\")\n",
|
||||||
|
"else:\n",
|
||||||
|
" device = torch.device(\"cpu\")\n",
|
||||||
"\n",
|
"\n",
|
||||||
"print(\"Device:\", device)"
|
"print(\"Device:\", device)"
|
||||||
]
|
]
|
||||||
|
|||||||
@@ -727,6 +727,8 @@
|
|||||||
" device = torch.device(\"mps\")\n",
|
" device = torch.device(\"mps\")\n",
|
||||||
" else:\n",
|
" else:\n",
|
||||||
" device = torch.device(\"cpu\")\n",
|
" device = torch.device(\"cpu\")\n",
|
||||||
|
"else:\n",
|
||||||
|
" device = torch.device(\"cpu\")\n",
|
||||||
"\n",
|
"\n",
|
||||||
"print(\"Device:\", device)\n",
|
"print(\"Device:\", device)\n",
|
||||||
"\n",
|
"\n",
|
||||||
|
|||||||
Reference in New Issue
Block a user