From 1354cb9ed102ec02bfb451fb934935e90a731f2e Mon Sep 17 00:00:00 2001
From: Bryce <github20210803@accounts.brycedrennan.com>
Date: Sat, 9 Sep 2023 13:21:53 -0700
Subject: [PATCH] build: add minimum package requirements

Should speed up dependency resolution.
---
 requirements-dev.txt |  2 +-
 setup.py             | 35 ++++++++++++++++++-----------------
 2 files changed, 19 insertions(+), 18 deletions(-)

diff --git a/requirements-dev.txt b/requirements-dev.txt
index 46b373e..8bac51d 100644
--- a/requirements-dev.txt
+++ b/requirements-dev.txt
@@ -2,7 +2,7 @@
 # This file is autogenerated by pip-compile with Python 3.10
 # by the following command:
 #
-#    pip-compile --output-file=requirements-dev.txt --resolver=backtracking requirements-dev.in setup.py
+#    pip-compile --output-file=requirements-dev.txt requirements-dev.in setup.py
 #
 aiohttp==3.8.5
     # via fsspec
diff --git a/setup.py b/setup.py
index dae5ca6..d3721e3 100644
--- a/setup.py
+++ b/setup.py
@@ -72,41 +72,42 @@ setup(
         ]
     },
     install_requires=[
-        "click",
-        "click-help-colors",
-        "click-shell",
+        "click>=8.0.0",
+        "click-help-colors>=0.9.1",
+        "click-shell>=2.0",
         "protobuf != 3.20.2, != 3.19.5",
-        "facexlib",
+        "facexlib>=0.2.1.1",
         "fairscale>=0.4.4",  # for vendored blip
-        "fastapi",
-        "ftfy",  # for vendored clip
-        # 2.0.0 produced garbage images on MacOS
+        "fastapi>=0.70.0",
+        "ftfy>=6.0.1",  # for vendored clip
+        # 2.0.0 produced garbage images on macOS
         "torch>=1.13.1,<2.0.0",
-        "numpy",
-        "tqdm",
-        "diffusers",
+        # https://numpy.org/neps/nep-0029-deprecation_policy.html
+        "numpy>=1.19.0,<1.26.0",
+        "tqdm>=4.64.0",
+        "diffusers>=0.3.0",
         "imageio>=2.9.0",
         "Pillow>=9.1.0",
-        "psutil",
+        "psutil>5.7.3",
         # 2.0.0 need to fix `ImportError: cannot import name 'rank_zero_only' from 'pytorch_lightning.utilities.distributed' `
         "pytorch-lightning>=1.4.2,<2.0.0",
         "omegaconf>=2.1.1",
-        "open-clip-torch",
-        "opencv-python",
+        "open-clip-torch>=2.0.0",
+        "opencv-python>=4.4.0.46",
         # need to migration to 2.0
         "pydantic<2.0.0",
-        "requests",
+        "requests>=2.28.1",
         "einops>=0.3.0",
-        "safetensors",
+        "safetensors>=0.2.1",
         # scipy is a sub dependency but v1.11 doesn't support python 3.8.  https://docs.scipy.org/doc/scipy/dev/toolchain.html#numpy
         "scipy<1.11",
         "timm>=0.4.12,!=0.9.0,!=0.9.1",  # for vendored blip
-        "torchdiffeq",
+        "torchdiffeq>=0.2.0",
         "transformers>=4.19.2",
         "torchmetrics>=0.6.0",
         "torchvision>=0.13.1",
         "kornia>=0.6",
-        "uvicorn",
+        "uvicorn>=0.16.0",
         "xformers>=0.0.16; sys_platform!='darwin' and platform_machine!='aarch64'",
     ],
     # torchvision doesn't support python 3.11 unless we switch to torch 2.0