diff --git a/pyproject.toml b/pyproject.toml index 8ea9aac..7312b76 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -10,6 +10,7 @@ dependencies = [ "pillow>=10.1.0", "jaxtyping>=0.2.23", "packaging>=23.2", + "numpy>=1.26.4", ] readme = "README.md" requires-python = ">= 3.10" diff --git a/requirements.lock b/requirements.lock index aeb101b..e2dcbb2 100644 --- a/requirements.lock +++ b/requirements.lock @@ -145,6 +145,7 @@ numpy==1.26.4 # via pandas # via pyarrow # via pywavelets + # via refiners # via torchvision # via transformers nvidia-cublas-cu12==12.1.3.1