xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 21:44:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Remove JAX constraints in Exporter (#8309)

Browse Source
This commit is contained in:
Glenn Jocher 2024-02-19 19:44:51 +01:00 committed by GitHub
parent d8b8b71866
commit e4a62f99f7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/ci.yaml vendored
View File

@ -113,8 +113,6 @@ jobs:
run: | run: |
python -m pip install --upgrade pip wheel python -m pip install --upgrade pip wheel
pip install -e ".[export]" "coverage[toml]" --extra-index-url https://download.pytorch.org/whl/cpu pip install -e ".[export]" "coverage[toml]" --extra-index-url https://download.pytorch.org/whl/cpu
# Fix SavedModel issue "partially initialized module 'jax' has no attribute 'version' (most likely due to a circular import)" in https://github.com/google/jax/discussions/14036
# pip install -U 'jax!=0.4.15' 'jaxlib!=0.4.15'
yolo export format=tflite imgsz=32 || true yolo export format=tflite imgsz=32 || true
- name: Check environment - name: Check environment
run: | run: |

3
ultralytics/engine/exporter.py
View File

@ -860,8 +860,7 @@ class Exporter:
@try_export @try_export
def export_tfjs(self, prefix=colorstr("TensorFlow.js:")): def export_tfjs(self, prefix=colorstr("TensorFlow.js:")):
"""YOLOv8 TensorFlow.js export.""" """YOLOv8 TensorFlow.js export."""
# JAX bug requiring install constraints in https://github.com/google/jax/issues/18978 check_requirements("tensorflowjs")
check_requirements(["jax<=0.4.21", "jaxlib<=0.4.21", "tensorflowjs"])
import tensorflow as tf import tensorflow as tf
import tensorflowjs as tfjs # noqa import tensorflowjs as tfjs # noqa