From b9b315a1e2af006c23da4cd88cf7f16015ea7432 Mon Sep 17 00:00:00 2001 From: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com> Date: Wed, 12 Feb 2025 20:38:13 +0800 Subject: [PATCH] Force protobuf>=5 for SavedModel export (#19206) Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com> --- ultralytics/engine/exporter.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index 65dba0b9..68d9f9d1 100644 --- 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 )