From 81997c61ae1c19379997a28e0d40a1f0f41e1170 Mon Sep 17 00:00:00 2001 From: Gregory Comer Date: Wed, 17 Apr 2024 14:21:22 -0700 Subject: [PATCH] Add command tests to pull CI --- .github/workflows/pull.yml | 2 +- .github/workflows/test_torchchat_commands.yml | 5 +++++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/.github/workflows/pull.yml b/.github/workflows/pull.yml index 193e5c7bd..9c7e10d03 100644 --- a/.github/workflows/pull.yml +++ b/.github/workflows/pull.yml @@ -173,4 +173,4 @@ jobs: echo "::group::Run inference" bash .ci/scripts/validate.sh "./checkpoints/${REPO_NAME}/model.pth" "cuda" "aoti" - echo "::endgroup::" + echo "::endgroup::" \ No newline at end of file diff --git a/.github/workflows/test_torchchat_commands.yml b/.github/workflows/test_torchchat_commands.yml index 86c4d8224..fdcc3f5d9 100644 --- a/.github/workflows/test_torchchat_commands.yml +++ b/.github/workflows/test_torchchat_commands.yml @@ -54,4 +54,9 @@ jobs: cat ./output_eager1 cat ./output_eager2 echo "Tests complete." + + - name: Test download + run: | + python torchchat.py download llama2 + python torchchat.py generate llama2 \ No newline at end of file