Add command tests to pull CI

pytorch · Apr 17, 2024 · 81997c6 · 81997c6
1 parent ac66e27
commit 81997c6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git 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::"
diff --git 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