From db125a20175384d75560dc9af7fb1100d67213fe Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 10 Mar 2024 13:49:37 +0100 Subject: [PATCH] Update export.py to `tensorflow<=2.15.1` (#12803) * Update export.py Signed-off-by: Glenn Jocher * Update export.py Signed-off-by: Glenn Jocher * Auto-format by https://ultralytics.com/actions --------- Signed-off-by: Glenn Jocher Co-authored-by: UltralyticsAssistant --- export.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/export.py b/export.py index 8fe7ce8fdce1..c660ce660aa5 100644 --- a/export.py +++ b/export.py @@ -405,7 +405,8 @@ def export_saved_model( try: import tensorflow as tf except Exception: - check_requirements(f"tensorflow{'' if torch.cuda.is_available() else '-macos' if MACOS else '-cpu'}") + check_requirements(f"tensorflow{'' if torch.cuda.is_available() else '-macos' if MACOS else '-cpu'}<=2.15.1") + import tensorflow as tf from tensorflow.python.framework.convert_to_constants import convert_variables_to_constants_v2