diff --git a/requirements-ocr.txt b/requirements-ocr.txt index dad26f4..767a9a6 100644 --- a/requirements-ocr.txt +++ b/requirements-ocr.txt @@ -1,3 +1,5 @@ torch transformers <= 4.30.2 ; python_version < '3.10' transformers >= 5 ; python_version >= '3.10' +tensorflow < 2.16 # for tensorflow-addons, so only needed in training +tf-keras < 2.16 # avoid keras 3 (also needs TF_USE_LEGACY_KERAS=1)