From 5e68f2efcffe825bbd30cae897c94fb696e6c30d Mon Sep 17 00:00:00 2001 From: Craig Pfeifer Date: Sun, 15 Dec 2024 14:27:24 -0500 Subject: [PATCH] fix checkpoint dir --- tests/test_readme.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_readme.py b/tests/test_readme.py index 8d7c68c81d..131d28de24 100644 --- a/tests/test_readme.py +++ b/tests/test_readme.py @@ -173,7 +173,7 @@ def test_continue_pretrain_model(tmp_path): @pytest.mark.dependency(depends=["test_download_model"]) def test_serve(): - CHECKPOINT_DIR = str("/Users/craig/code/litgpt/checkpoints" / REPO_ID) + CHECKPOINT_DIR = str("checkpoints" / REPO_ID) run_command = [ "litgpt", "serve", "--devices","0", str(CHECKPOINT_DIR) ]