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

User argpars utils to show default args on command line

Browse Source
This commit is contained in:
rasbt
2026-03-01 20:15:21 -06:00
parent c079904491
commit 4612d20fa8
27 changed files with 30 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
ch05/01_main-chapter-code/gpt_generate.py
View File

@@ -253,16 +253,16 @@ def main(gpt_config, input_prompt, model_size, device):
if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Generate text with a pretrained GPT-2 model.")
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter, description="Generate text with a pretrained GPT-2 model.")
parser.add_argument(
"--prompt",
default="Every effort moves you",
help="Prompt text used to seed the generation (default matches the script's built-in prompt)."
help="Prompt text used to seed the generation."
)
parser.add_argument(
"--device",
default="cpu",
help="Device for running inference, e.g., cpu, cuda, mps, or auto. Defaults to cpu."
help="Device for running inference, e.g., cpu, cuda, mps, or auto."
)
args = parser.parse_args()