pull/17749/head
Laughing-q 2 months ago
parent ad4d6828ae
commit afa316b215
  1. 2
      .github/workflows/ci.yaml
  2. 2
      pyproject.toml

@ -110,7 +110,7 @@ jobs:
- name: Install requirements
shell: bash # for Windows compatibility
run: |
uv pip install --system -e ".[export]" "coverage[toml]" --extra-index-url https://download.pytorch.org/whl/cpu
uv pip install --prerelease=allow --system -e ".[export]" "coverage[toml]" --extra-index-url https://download.pytorch.org/whl/cpu
- name: Check environment
run: |
yolo checks

@ -101,7 +101,7 @@ export = [
"scikit-learn>=1.3.2; platform_system != 'Windows' and python_version <= '3.11'", # CoreML k-means quantization
"openvino>=2024.0.0", # OpenVINO export
"tensorflow>=2.0.0", # TF bug https://github.com/ultralytics/ultralytics/issues/5161
"tensorflowjs>=4.5.0,<4.11.0", # TF.js export, automatically installs tensorflow
"tensorflowjs>=4.5.0", # TF.js export, automatically installs tensorflow
"tensorstore>=0.1.63; platform_machine == 'aarch64' and python_version >= '3.9'", # for TF Raspberry Pi exports
"keras", # not installed automatically by tensorflow>=2.16
"flatbuffers>=23.5.26,<100; platform_machine == 'aarch64'", # update old 'flatbuffers' included inside tensorflow package

Loading…
Cancel
Save