[tnx][ci] add resnet test w/o requirements.txt (#1817)

deepjavalibrary · Apr 26, 2024 · 57d32bc · 57d32bc
1 parent f6b773d
commit 57d32bc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.github/workflows/llm_inf2_integration.yml b/.github/workflows/llm_inf2_integration.yml
@@ -74,7 +74,7 @@ jobs:
         working-directory: tests/integration
         run: |
           ./launch_container.sh deepjavalibrary/djl-serving:$DJLSERVING_DOCKER_TAG $PWD/models pytorch-inf2-1 \
-          serve -m test::Python:nc0=file:/opt/ml/model/resnet18_inf2_2_4.tar.gz
+          serve -m test::Python:nc0=file:/opt/ml/model/resnet18_no_reqs_inf2_2_4.tar.gz
           ./test_client.sh image/jpg models/kitten.jpg
           docker rm -f $(docker ps -aq)
       - name: Test transformers-neuronx gpt2 with handler

diff --git a/tests/integration/download_models.sh b/tests/integration/download_models.sh
@@ -24,6 +24,7 @@ aarch_models_urls=(
 
 inf2_models_urls=(
   "https://resources.djl.ai/test-models/pytorch/resnet18_inf2_2_4.tar.gz"
+  "https://resources.djl.ai/test-models/pytorch/resnet18_no_reqs_inf2_2_4.tar.gz"
 )
 
 download() {