diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index 0beb38b9..b1a52b0f 100644 --- a/ultralytics/engine/exporter.py +++ b/ultralytics/engine/exporter.py @@ -73,6 +73,7 @@ from ultralytics.nn.tasks import DetectionModel, SegmentationModel, WorldModel from ultralytics.utils import ( ARM64, DEFAULT_CFG, + IS_JETSON, LINUX, LOGGER, MACOS, @@ -749,7 +750,7 @@ class Exporter: "sng4onnx>=1.0.1", "onnxsim>=0.4.33", "onnx_graphsurgeon>=0.3.26", - "tflite_support", + "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", ),