Remove JAX constraints in Exporter (#8309)
This commit is contained in:
parent
d8b8b71866
commit
e4a62f99f7
2 changed files with 1 additions and 4 deletions
2
.github/workflows/ci.yaml
vendored
2
.github/workflows/ci.yaml
vendored
|
|
@ -113,8 +113,6 @@ jobs:
|
|||
run: |
|
||||
python -m pip install --upgrade pip wheel
|
||||
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
|
||||
- name: Check environment
|
||||
run: |
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue