diff --git a/pyproject.toml b/pyproject.toml
index c3c2adf..b568bd6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -10,6 +10,7 @@ dependencies = [
     "pillow>=10.1.0",
     "jaxtyping>=0.2.23",
     "packaging>=23.2",
+    "ruff>=0.2.0",
 ]
 readme = "README.md"
 requires-python = ">= 3.10"
@@ -86,6 +87,9 @@ line-length = 120
 
 [tool.ruff]
 src = ["src"] # see https://docs.astral.sh/ruff/settings/#src
+line-length = 120
+
+[tool.ruff.lint]
 select = [
     "I", # isort
 ]
@@ -93,7 +97,6 @@ ignore = [
     "F722", # forward-annotation-syntax-error, because of Jaxtyping
     "E731", # do-not-assign-lambda
 ]
-line-length = 120
 
 [tool.ruff.lint.isort]
 # Allow this kind of import on a single line:
diff --git a/requirements.lock b/requirements.lock
index 0d4d992..f8770b6 100644
--- a/requirements.lock
+++ b/requirements.lock
@@ -17,12 +17,12 @@ attrs==23.2.0
 babel==2.14.0
 bitsandbytes==0.42.0
 black==24.1.1
-certifi==2023.11.17
+certifi==2024.2.2
 charset-normalizer==3.3.2
 click==8.1.7
 colorama==0.4.6
 datasets==2.16.1
-diffusers==0.25.1
+diffusers==0.26.1
 dill==0.3.7
 docker-pycreds==0.4.0
 filelock==3.13.1
@@ -50,7 +50,7 @@ mkdocs-material-extensions==1.3.1
 mkdocstrings==0.24.0
 mkdocstrings-python==1.8.0
 mpmath==1.3.0
-multidict==6.0.4
+multidict==6.0.5
 multiprocess==0.70.15
 mypy-extensions==1.0.0
 networkx==3.2.1
@@ -85,12 +85,13 @@ pydantic-core==2.16.1
 pygments==2.17.2
 pymdown-extensions==10.7
 python-dateutil==2.8.2
-pytz==2023.4
+pytz==2024.1
 pywavelets==1.5.0
 pyyaml==6.0.1
 pyyaml-env-tag==0.1
 regex==2023.12.25
 requests==2.31.0
+ruff==0.2.0
 safetensors==0.4.2
 scipy==1.12.0
 segment-anything-py==1.0