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

Add PyPI package (#576)

Browse Source 
* Add PyPI package

* fixes

* fixes
This commit is contained in:
Sebastian Raschka
2025-03-23 19:28:49 -05:00
committed by GitHub
parent 7757c3d308
commit c21bfe4a23
52 changed files with 2173 additions and 2585 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1035,6 +1035,12 @@
"source": [
"from gpt_download import download_and_load_gpt2\n",
"from previous_chapters import GPTModel, load_weights_into_gpt\n",
"# If the `previous_chapters.py` file is not available locally,\n",
"# you can import it from the `llms-from-scratch` PyPI package.\n",
"# For details, see: https://github.com/rasbt/LLMs-from-scratch/tree/main/pkg\n",
"# E.g.,\n",
"# from llms_from_scratch.ch04 import GPTModel\n",
"# from llms_from_scratch.ch05 import download_and_load_gpt2, load_weights_into_gpt\n",
"\n",
"model_size = CHOOSE_MODEL.split(\" \")[-1].lstrip(\"(\").rstrip(\")\")\n",
"settings, params = download_and_load_gpt2(model_size=model_size, models_dir=\"gpt2\")\n",
@@ -1075,6 +1081,13 @@
" token_ids_to_text\n",
")\n",
"\n",
"# Alternatively:\n",
"# from llms_from_scratch.ch05 import (\n",
"# generate_text_simple,\n",
"# text_to_token_ids,\n",
"# token_ids_to_text\n",
"# )\n",
"\n",
"\n",
"text_1 = \"Every effort moves you\"\n",
"\n",