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

Set up basic test gh worklows (#79)

Browse Source 
* Set up basic test gh worklows

* update file paths

* env check

* add env check

* Update requirements.txt

* simplify

* upd
This commit is contained in:
Sebastian Raschka
2024-03-18 11:58:37 -05:00
committed by GitHub
parent 9d6da22ebb
commit ca96abac8a
13 changed files with 50241 additions and 110 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
ch04/01_main-chapter-code/gpt.py
View File

@@ -235,8 +235,7 @@ def generate_text_simple(model, idx, max_new_tokens, context_size):
return idx
if __name__ == "__main__":
def main():
GPT_CONFIG_124M = {
"vocab_size": 50257, # Vocabulary size
"ctx_len": 1024, # Context length
@@ -274,3 +273,7 @@ if __name__ == "__main__":
print("\nOutput:", out)
print("Output length:", len(out[0]))
print("Output text:", decoded_text)
if __name__ == "__main__":
main()