Skip to content

Florence2 workflows block #1551

Florence2 workflows block

Florence2 workflows block #1551

name: INTEGRATION TESTS - inference models
on:
pull_request:
branches: [main]
push:
branches: [ main ]
workflow_dispatch:
jobs:
build-dev-test:
runs-on: ubuntu-latest
timeout-minutes: 15
strategy:
matrix:
python-version: ["3.8", "3.9", "3.10", "3.11"]
steps:
- name: 🛎️ Checkout
uses: actions/checkout@v4
with:
ref: ${{ github.head_ref }}
- name: 🐍 Set up Python ${{ matrix.python-version }}
uses: actions/setup-python@v5
with:
python-version: ${{ matrix.python-version }}
check-latest: true
- name: 🚧 Install GDAL OS library
run: sudo apt-get update && sudo apt-get install libgdal-dev
- name: 📦 Install dependencies
run: |
python -m pip install --upgrade pip
pip install --upgrade setuptools
pip install --extra-index-url https://download.pytorch.org/whl/cpu -r requirements/_requirements.txt -r requirements/requirements.cpu.txt -r requirements/requirements.sdk.http.txt -r requirements/requirements.test.unit.txt -r requirements/requirements.http.txt -r requirements/requirements.yolo_world.txt -r requirements/requirements.sam.txt -r requirements/requirements.transformers.txt
- name: 🧪 Integration Tests of Inference models
timeout-minutes: 45
run: MAX_BATCH_SIZE=6 python -m pytest tests/inference/models_predictions_tests