Skip to content

Commit

Permalink
Merge pull request #632 from PINTO0309/flat_verup
Browse files Browse the repository at this point in the history
flatbuffers #631
  • Loading branch information
PINTO0309 authored May 16, 2024
2 parents 7db688e + 062a07f commit 68de5a2
Show file tree
Hide file tree
Showing 4 changed files with 11 additions and 6 deletions.
1 change: 1 addition & 0 deletions .github/workflows/test-models.yml
Original file line number Diff line number Diff line change
Expand Up @@ -106,6 +106,7 @@ jobs:
pip install onnxruntime==1.17.1
pip install ml_dtypes==0.3.2
pip install tf-keras~=2.16
pip install flatbuffers>=23.5.26
pip install -e .
- name: Download models
run: |
Expand Down
3 changes: 2 additions & 1 deletion Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,8 @@ RUN pip install pip -U \
&& pip install psutil==5.9.5 \
&& pip install onnxruntime==1.17.1 \
&& pip install ml_dtypes==0.3.2 \
&& pip install tf-keras~=2.16
&& pip install tf-keras~=2.16 \
&& pip install flatbuffers>=23.5.26

# Re-release flatc with some customizations of our own to address
# the lack of arithmetic precision of the quantization parameters
Expand Down
11 changes: 7 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -241,6 +241,7 @@ Video speed is adjusted approximately 50 times slower than actual speed.
- psutil==5.9.5
- ml_dtypes==0.3.2
- flatbuffers-compiler (Optional, Only when using the `-coion` option. Executable file named `flatc`.)
- flatbuffers>=23.5.26
```bash
# Custom flatc v23.5.26 binary for Ubuntu 20.04+
# https://github.com/PINTO0309/onnx2tf/issues/196
Expand Down Expand Up @@ -270,15 +271,15 @@ Video speed is adjusted approximately 50 times slower than actual speed.
docker run --rm -it \
-v `pwd`:/workdir \
-w /workdir \
ghcr.io/pinto0309/onnx2tf:1.21.2
ghcr.io/pinto0309/onnx2tf:1.21.3

or

# Authentication is not required for pulls from Docker Hub.
docker run --rm -it \
-v `pwd`:/workdir \
-w /workdir \
docker.io/pinto0309/onnx2tf:1.21.2
docker.io/pinto0309/onnx2tf:1.21.3

or

Expand All @@ -296,7 +297,8 @@ Video speed is adjusted approximately 50 times slower than actual speed.
&& pip install -U h5py==3.11.0 \
&& pip install -U psutil==5.9.5 \
&& pip install -U ml_dtypes==0.3.2 \
&& pip install -U tf-keras~=2.16
&& pip install -U tf-keras~=2.16 \
&& pip install flatbuffers>=23.5.26

or

Expand Down Expand Up @@ -327,7 +329,8 @@ or
&& pip install -U h5py==3.11.0 \
&& pip install -U psutil==5.9.5 \
&& pip install -U ml_dtypes==0.3.2 \
&& pip install -U tf-keras~=2.16
&& pip install -U tf-keras~=2.16 \
&& pip install flatbuffers>=23.5.26
```

### 2. Run test
Expand Down
2 changes: 1 addition & 1 deletion onnx2tf/__init__.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
from onnx2tf.onnx2tf import convert, main

__version__ = '1.21.2'
__version__ = '1.21.3'

0 comments on commit 68de5a2

Please sign in to comment.