diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index 240245cf..20c7d3b7 100644 --- a/ultralytics/engine/exporter.py +++ b/ultralytics/engine/exporter.py @@ -722,6 +722,7 @@ class Exporter: "onnxsim>=0.4.33", "onnx_graphsurgeon>=0.3.26", "tflite_support", + "flatbuffers>=23.5.26", # update old 'flatbuffers' included inside tensorflow package "onnxruntime-gpu" if cuda else "onnxruntime", ), cmds="--extra-index-url https://pypi.ngc.nvidia.com",