schihei/llama2.c
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

oops i should probably call this function lol

Browse Source
This commit is contained in:
Andrej Karpathy
2023-08-13 23:48:01 +00:00
parent 86325bf7e8
commit c970f69334
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test_all.py
+2
View File
@@ -42,6 +42,7 @@ expected_stdout = b'Once upon a time, there was a little girl named Lily. She lo
def test_runc(): def test_runc():
""" Forwards a model against a known-good desired outcome in run.c for 200 steps""" """ Forwards a model against a known-good desired outcome in run.c for 200 steps"""
attempt_download_files()
model_path = os.path.join(test_ckpt_dir, "stories260K.bin") model_path = os.path.join(test_ckpt_dir, "stories260K.bin")
tokenizer_path = os.path.join(test_ckpt_dir, "tok512.bin") tokenizer_path = os.path.join(test_ckpt_dir, "tok512.bin")
@@ -56,6 +57,7 @@ def test_runc():
def test_python(): def test_python():
""" Forwards a model against a known-good desired outcome in sample.py for 200 steps""" """ Forwards a model against a known-good desired outcome in sample.py for 200 steps"""
attempt_download_files()
device = "cpu" # stories260K is small enough to just breeze through it on CPU device = "cpu" # stories260K is small enough to just breeze through it on CPU
checkpoint = os.path.join(test_ckpt_dir, "stories260K.pt") checkpoint = os.path.join(test_ckpt_dir, "stories260K.pt")