diff --git a/setup.py b/setup.py
index b58a56a..9288a1b 100644
--- a/setup.py
+++ b/setup.py
@@ -224,7 +224,7 @@ def get_model_zoo_configs() -> List[str]:
         "dataclasses; python_version<'3.7'",
         "omegaconf>=2.1.0rc1",
         "hydra-core>=1.1.0rc1",
-        "black==21.4b2",
+        "black==24.3.0",
         # When adding to the list, may need to update docs/requirements.txt
         # or add mock in docs/conf.py
     ],