Force protobuf>=5 for SavedModel export (#19206)

Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
2025-02-12 20:38:13 +08:00 · 2025-02-12 20:38:13 +08:00 · b9b315a1e2
commit b9b315a1e2
parent 1bb46cda4b
1 changed files with 1 additions and 0 deletions
--- a/ultralytics/engine/exporter.py
+++ b/ultralytics/engine/exporter.py
@ -986,6 +986,7 @@ class Exporter:
                "tflite_support<=0.4.3" if IS_JETSON else "tflite_support",  # fix ImportError 'GLIBCXX_3.4.29'
                "flatbuffers>=23.5.26,<100",  # update old 'flatbuffers' included inside tensorflow package
                "onnxruntime-gpu" if cuda else "onnxruntime",
+                "protobuf>=5",  # tflite_support pins <=4 but >=5 works
            ),
            cmds="--extra-index-url https://pypi.ngc.nvidia.com",  # onnx_graphsurgeon only on NVIDIA
        )