Update export.py to tensorflow<=2.15.1 (#12803)

* Update export.py Signed-off-by: Glenn Jocher <[email protected]> * Update export.py Signed-off-by: Glenn Jocher <[email protected]> * Auto-format by https://ultralytics.com/actions --------- Signed-off-by: Glenn Jocher <[email protected]> Co-authored-by: UltralyticsAssistant <[email protected]>
Kovelja009 · Mar 10, 2024 · db125a2 · db125a2
1 parent a428ff7
commit db125a2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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