yujunjun/LLMs-from-scratch
1
0
Fork 0
mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2025-08-29 11:00:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove duplicated cell (#212)

Browse Source 
* add a suggestion since code snippet has been repeated.

* remove duplicated cell

---------

Co-authored-by: Shuyib <benmainye@gmail.com>
This commit is contained in:
Sebastian Raschka 2024-06-15 12:48:34 -05:00 committed by GitHub
parent a796b9d657
commit fcf8bcab0d
1 changed files with 1 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -390,32 +390,6 @@
"print(\"Token IDs:\\n\", token_ids)" "print(\"Token IDs:\\n\", token_ids)"
] ]
}, },
{
"cell_type": "code",
"execution_count": 7,
"id": "f08e4efc-e897-4d35-91f7-9a77b6dde953",
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Token IDs:\n",
" tensor([[[16657],\n",
" [ 339],\n",
" [42826]],\n",
"\n",
" [[49906],\n",
" [29669],\n",
" [41751]]])\n"
]
}
],
"source": [
"token_ids = torch.argmax(probas, dim=-1, keepdim=True)\n",
"print(\"Token IDs:\\n\", token_ids)"
]
},
{ {
"cell_type": "markdown", "cell_type": "markdown",
"id": "cee4072c-21ed-4df7-8721-dd2535362573", "id": "cee4072c-21ed-4df7-8721-dd2535362573",
@ -2480,7 +2454,7 @@
"name": "python", "name": "python",
"nbconvert_exporter": "python", "nbconvert_exporter": "python",
"pygments_lexer": "ipython3", "pygments_lexer": "ipython3",
"version": "3.11.4" "version": "3.10.6"
} }
}, },
"nbformat": 4, "nbformat": 4,