From 18c4d40fb29c9b11880e8e6ce0c8cbd69fc5d528 Mon Sep 17 00:00:00 2001 From: Bryce Date: Sat, 21 Sep 2024 18:51:26 -0700 Subject: [PATCH] build: update requirements --- requirements-dev.txt | 180 ++++++++++++++++++++++++------------------- setup.py | 5 +- 2 files changed, 104 insertions(+), 81 deletions(-) diff --git a/requirements-dev.txt b/requirements-dev.txt index 11f68013..1873751f 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -4,19 +4,21 @@ # # pip-compile --output-file=requirements-dev.txt requirements-dev.in setup.py # -annotated-types==0.6.0 +accelerate==0.34.2 + # via imaginAIry (setup.py) +annotated-types==0.7.0 # via pydantic antlr4-python3-runtime==4.9.3 # via omegaconf -anyio==4.3.0 +anyio==4.6.0 # via # httpx # starlette -babel==2.14.0 +babel==2.16.0 # via mkdocs-material -build==1.2.1 +build==1.2.2 # via pip-tools -certifi==2024.2.2 +certifi==2024.8.30 # via # httpcore # httpx @@ -41,37 +43,37 @@ colorama==0.4.6 # via # griffe # mkdocs-material -coverage==7.4.4 +coverage==7.6.1 # via -r requirements-dev.in -diffusers==0.27.2 +diffusers==0.30.3 # via imaginAIry (setup.py) -einops==0.7.0 +einops==0.8.0 # via # imaginAIry (setup.py) # spandrel -exceptiongroup==1.2.0 +exceptiongroup==1.2.2 # via # anyio # pytest -fastapi==0.110.1 +fastapi==0.115.0 # via imaginAIry (setup.py) -filelock==3.13.4 +filelock==3.16.1 # via # diffusers # huggingface-hub # torch # transformers -fsspec==2024.3.1 +fsspec==2024.9.0 # via # huggingface-hub # torch -ftfy==6.2.0 +ftfy==6.2.3 # via # imaginAIry (setup.py) # open-clip-torch ghp-import==2.1.0 # via mkdocs -griffe==0.42.2 +griffe==1.3.1 # via mkdocstrings-python h11==0.14.0 # via @@ -79,37 +81,38 @@ h11==0.14.0 # uvicorn httpcore==1.0.5 # via httpx -httpx==0.27.0 +httpx==0.27.2 # via -r requirements-dev.in -huggingface-hub==0.22.2 +huggingface-hub==0.25.0 # via + # accelerate # diffusers # open-clip-torch # timm # tokenizers # transformers -idna==3.7 +idna==3.10 # via # anyio # httpx # requests -importlib-metadata==7.1.0 +importlib-metadata==8.5.0 # via diffusers iniconfig==2.0.0 # via pytest -jaxtyping==0.2.28 +jaxtyping==0.2.34 # via imaginAIry (setup.py) -jinja2==3.1.3 +jinja2==3.1.4 # via # mkdocs # mkdocs-material # mkdocstrings # torch -kornia==0.7.2 +kornia==0.7.3 # via imaginAIry (setup.py) -kornia-rs==0.1.3 +kornia-rs==0.1.5 # via kornia -markdown==3.6 +markdown==3.7 # via # mkdocs # mkdocs-autorefs @@ -124,53 +127,65 @@ markupsafe==2.1.5 # mkdocs-autorefs # mkdocstrings mergedeep==1.3.4 - # via mkdocs -mkdocs==1.5.3 + # via + # mkdocs + # mkdocs-get-deps +mkdocs==1.6.1 # via # mkdocs-autorefs # mkdocs-material # mkdocstrings -mkdocs-autorefs==1.0.1 - # via mkdocstrings +mkdocs-autorefs==1.2.0 + # via + # mkdocstrings + # mkdocstrings-python mkdocs-click==0.8.1 # via -r requirements-dev.in -mkdocs-material==9.5.18 +mkdocs-get-deps==0.2.0 + # via mkdocs +mkdocs-material==9.5.36 # via -r requirements-dev.in mkdocs-material-extensions==1.3.1 # via mkdocs-material -mkdocstrings[python]==0.24.3 +mkdocstrings[python]==0.26.1 # via # -r requirements-dev.in # mkdocstrings-python -mkdocstrings-python==1.9.2 +mkdocstrings-python==1.11.1 # via mkdocstrings mpmath==1.3.0 # via sympy -mypy==1.9.0 +mypy==1.11.2 # via -r requirements-dev.in mypy-extensions==1.0.0 # via mypy networkx==3.3 # via torch +ninja==1.11.1.1 + # via optimum-quanto numpy==1.24.4 # via # -c tests/constraints.txt + # accelerate # diffusers # imaginAIry (setup.py) - # jaxtyping # opencv-python + # optimum-quanto # scipy # spandrel # torchvision # transformers omegaconf==2.3.0 # via imaginAIry (setup.py) -open-clip-torch==2.24.0 +open-clip-torch==2.26.1 + # via imaginAIry (setup.py) +opencv-python==4.10.0.84 # via imaginAIry (setup.py) -opencv-python==4.9.0.80 +optimum-quanto==0.2.4 # via imaginAIry (setup.py) -packaging==24.0 +packaging==24.1 # via + # accelerate # build # huggingface-hub # kornia @@ -178,54 +193,54 @@ packaging==24.0 # pytest # pytest-sugar # transformers -paginate==0.5.6 +paginate==0.5.7 # via mkdocs-material pathspec==0.12.1 # via mkdocs -pillow==10.3.0 +pillow==10.4.0 # via # diffusers # imaginAIry (setup.py) # torchvision pip-tools==7.4.1 # via -r requirements-dev.in -platformdirs==4.2.0 +platformdirs==4.3.6 # via - # mkdocs + # mkdocs-get-deps # mkdocstrings -pluggy==1.4.0 +pluggy==1.5.0 # via pytest -protobuf==5.26.1 +protobuf==5.28.2 + # via imaginAIry (setup.py) +psutil==6.0.0 # via + # accelerate # imaginAIry (setup.py) - # open-clip-torch -psutil==5.9.8 - # via imaginAIry (setup.py) -pydantic==2.7.0 +pydantic==2.9.2 # via # fastapi # imaginAIry (setup.py) -pydantic-core==2.18.1 +pydantic-core==2.23.4 # via pydantic -pygments==2.17.2 +pygments==2.18.0 # via mkdocs-material -pymdown-extensions==10.7.1 +pymdown-extensions==10.9 # via # mkdocs-material # mkdocstrings -pyparsing==3.1.2 +pyparsing==3.1.4 # via imaginAIry (setup.py) -pyproject-hooks==1.0.0 +pyproject-hooks==1.1.0 # via # build # pip-tools -pytest==8.1.1 +pytest==8.3.3 # via # -r requirements-dev.in # pytest-asyncio # pytest-randomly # pytest-sugar -pytest-asyncio==0.23.6 +pytest-asyncio==0.24.0 # via -r requirements-dev.in pytest-randomly==3.15.0 # via -r requirements-dev.in @@ -233,10 +248,12 @@ pytest-sugar==1.0.0 # via -r requirements-dev.in python-dateutil==2.9.0.post0 # via ghp-import -pyyaml==6.0.1 +pyyaml==6.0.2 # via + # accelerate # huggingface-hub # mkdocs + # mkdocs-get-deps # omegaconf # pymdown-extensions # pyyaml-env-tag @@ -245,13 +262,13 @@ pyyaml==6.0.1 # transformers pyyaml-env-tag==0.1 # via mkdocs -regex==2024.4.16 +regex==2024.9.11 # via # diffusers # mkdocs-material # open-clip-torch # transformers -requests==2.31.0 +requests==2.32.3 # via # diffusers # huggingface-hub @@ -259,88 +276,91 @@ requests==2.31.0 # mkdocs-material # responses # transformers -responses==0.25.0 +responses==0.25.3 # via -r requirements-dev.in -ruff==0.3.7 +ruff==0.6.7 # via -r requirements-dev.in -safetensors==0.4.3 +safetensors==0.4.5 # via + # accelerate # diffusers # imaginAIry (setup.py) + # optimum-quanto # spandrel # timm # transformers -scipy==1.13.0 +scipy==1.14.1 # via # imaginAIry (setup.py) # torchdiffeq sentencepiece==0.2.0 - # via open-clip-torch + # via imaginAIry (setup.py) six==1.16.0 # via python-dateutil sniffio==1.3.1 # via # anyio # httpx -spandrel==0.3.1 +spandrel==0.4.0 # via imaginAIry (setup.py) -starlette==0.37.2 +starlette==0.38.5 # via fastapi -sympy==1.12 +sympy==1.13.3 # via torch termcolor==2.4.0 # via # imaginAIry (setup.py) # pytest-sugar -timm==0.9.16 +timm==1.0.9 # via # imaginAIry (setup.py) # open-clip-torch -tokenizers==0.15.2 +tokenizers==0.19.1 # via transformers tomli==2.0.1 # via # build # mypy # pip-tools - # pyproject-hooks # pytest -torch==2.2.2 +torch==2.4.1 # via + # accelerate # imaginAIry (setup.py) # kornia # open-clip-torch + # optimum-quanto # spandrel # timm # torchdiffeq # torchvision -torchdiffeq==0.2.3 +torchdiffeq==0.2.4 # via imaginAIry (setup.py) -torchvision==0.17.2 +torchvision==0.19.1 # via # imaginAIry (setup.py) # open-clip-torch # spandrel # timm -tqdm==4.66.2 +tqdm==4.66.5 # via # huggingface-hub # imaginAIry (setup.py) # open-clip-torch # transformers -transformers==4.39.3 +transformers==4.44.2 # via imaginAIry (setup.py) typeguard==2.13.3 # via jaxtyping -types-pillow==10.2.0.20240415 +types-pillow==10.2.0.20240822 # via -r requirements-dev.in -types-psutil==5.9.5.20240316 +types-psutil==6.0.0.20240901 # via -r requirements-dev.in -types-requests==2.31.0.20240406 +types-requests==2.32.0.20240914 # via -r requirements-dev.in types-tqdm==4.66.0.20240417 # via -r requirements-dev.in -typing-extensions==4.11.0 +typing-extensions==4.12.2 # via # anyio # fastapi @@ -351,22 +371,22 @@ typing-extensions==4.11.0 # spandrel # torch # uvicorn -urllib3==2.2.1 +urllib3==2.2.3 # via # requests # responses # types-requests -uvicorn==0.29.0 +uvicorn==0.30.6 # via imaginAIry (setup.py) -watchdog==4.0.0 +watchdog==5.0.2 # via mkdocs wcwidth==0.2.13 # via ftfy -wheel==0.43.0 +wheel==0.44.0 # via # -r requirements-dev.in # pip-tools -zipp==3.18.1 +zipp==3.20.2 # via importlib-metadata # The following packages are considered to be unsafe in a requirements file: diff --git a/setup.py b/setup.py index 9bfa7c90..606c9257 100644 --- a/setup.py +++ b/setup.py @@ -84,7 +84,7 @@ def get_git_revision_hash() -> str: # https://numpy.org/neps/nep-0029-deprecation_policy.html "numpy>=1.22.0", "tqdm>=4.64.0", - "diffusers>=0.3.0", + "diffusers>=0.30.3", "Pillow>=9.1.0", "psutil>5.7.3", "omegaconf>=2.1.1", @@ -110,6 +110,9 @@ def get_git_revision_hash() -> str: "uvicorn>=0.16.0", "spandrel>=0.1.8", # "xformers>=0.0.22; sys_platform!='darwin' and platform_machine!='aarch64'", + "optimum-quanto>=0.2.4", # for flux quantization + "sentencepiece>=0.2.0", + "accelerate>=0.24.0", ], # don't specify maximum python versions as it can cause very long dependency resolution issues as the resolver # goes back to older versions of packages that didn't specify a maximum