diff --git a/.github/workflows/ci-testing.yml b/.github/workflows/ci-testing.yml index 2ffdd68..390591b 100644 --- a/.github/workflows/ci-testing.yml +++ b/.github/workflows/ci-testing.yml @@ -33,7 +33,7 @@ jobs: - uses: astral-sh/setup-uv@v6 - name: Install requirements run: | - uv pip install --system -r requirements.txt coremltools openvino-dev tensorflow --extra-index-url https://download.pytorch.org/whl/cpu --index-strategy unsafe-best-match + uv pip install --system -r requirements.txt coremltools openvino-dev "tensorflow<=2.19.0" --extra-index-url https://download.pytorch.org/whl/cpu --index-strategy unsafe-best-match yolo checks uv pip list - name: Benchmark DetectionModel diff --git a/pyproject.toml b/pyproject.toml index 3ef7a1f..edc506a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -98,7 +98,7 @@ export = [ "onnx>=1.12.0", # ONNX export "coremltools>=7.0; platform_system != 'Windows'", # CoreML only supported on macOS and Linux "openvino-dev>=2023.0", # OpenVINO export - "tensorflow>=2.0.0", # TF bug https://github.com/ultralytics/ultralytics/issues/5161 + "tensorflow>=2.0.0,<=2.19.0", # TF bug https://github.com/ultralytics/ultralytics/issues/5161 "tensorflowjs>=3.9.0", # TF.js export, automatically installs tensorflow ] # tensorflow>=2.4.1,<=2.13.1 # TF exports (-cpu, -aarch64, -macos)