diff --git a/setup.cfg b/setup.cfg
index f1cab47f7f..1911f3b328 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -31,6 +31,7 @@ install_requires =
     qkeras
     tabulate
     tensorflow
+    tensorflow-model-optimization<=0.7.5
 python_requires = >=3.8
 include_package_data = True
 scripts = scripts/hls4ml