Merge pull request #608 from containers/fix-deter-resnet-ms

allowing pip to install requirements

.github/workflows/model_servers.yaml

@@ -6,12 +6,14 @@ on:
       - main
     paths:
       - 'model_servers/**'
+      - 'models/Makefile'
       - .github/workflows/model_servers.yaml
   push:
     branches:
       - main
     paths:
       - 'model_servers/**'
+      - 'models/Makefile'
       - .github/workflows/model_servers.yaml
 
   workflow_dispatch:
@@ -83,6 +85,11 @@ jobs:
         run: |
           sudo apt-get update
           sudo apt-get install -y qemu-user-static
+
+      - name: Set up Python
+        uses: actions/setup-python@v5.1.0
+        with:
+          python-version: '3.11'
 
       - name: Build Image
         id: build_image
@@ -98,11 +105,6 @@ jobs:
         working-directory: ./models
         run: make download-model-${{ matrix.model }}
 
-      - name: Set up Python
-        uses: actions/setup-python@v5.1.0
-        with:
-          python-version: '3.11'
-
       - name: Install python dependencies
         working-directory: ./model_servers/${{ matrix.directory }}/
         run: make install

model_servers/object_detection_python/src/requirements-unlocked.txt

@@ -4,4 +4,5 @@ pydantic
 requests
 transformers
 torch
-uvicorn
+uvicorn
+timm

model_servers/object_detection_python/src/requirements.txt

@@ -7,7 +7,7 @@ dnspython==2.6.1
 email_validator==2.2.0
 fastapi==0.111.0
 fastapi-cli==0.0.4
-filelock==3.15.3
+filelock==3.15.4
 fsspec==2024.6.0
 h11==0.14.0
 httpcore==1.0.5
@@ -39,8 +39,10 @@ shellingham==1.5.4
 sniffio==1.3.1
 starlette==0.37.2
 sympy==1.12.1
+timm==1.0.7
 tokenizers==0.19.1
 torch==2.3.1
+torchvision==0.18.1
 tqdm==4.66.4
 transformers==4.41.2
 typer==0.12.3

models/Makefile

@@ -40,7 +40,7 @@ download-model-mistral-code:
 
 .PHONY: download-model-facebook-detr-resnet-101
 download-model-facebook-detr-resnet-101:
-	pip install -r ../convert_models/requirements.txt
+	python3 -m pip install -r ../convert_models/requirements.txt
 	cd ../convert_models/ && \
 	python3 download_huggingface.py -m facebook/detr-resnet-101
 	cp -r ../convert_models/converted_models/facebook ./