Add command tests to pull CI

pytorch · Apr 17, 2024 · dc728a3 · dc728a3
1 parent 81b74a5
commit dc728a3
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 mistral-7b-instruct
+          python torchchat.py generate mistral-7b-instruct