update requirements

sdbds · Sep 28, 2024 · 8922d1b · 8922d1b
1 parent 7578a9e
commit 8922d1b
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 78 deletions.
diff --git a/requirements-uv.txt b/requirements-uv.txt
@@ -1,30 +1,26 @@
 # This file was autogenerated by uv via the following command:
-#    uv pip compile requirements.txt -o requirements-uv.txt
+#    uv pip compile ./requirements.txt -o requirements-uv.txt -i https://pypi.org/simple --index-strategy unsafe-best-match
 -e .
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 absl-py==2.1.0
     # via
     #   keras
     #   tensorboard
     #   tensorflow-intel
 accelerate==0.33.0
     # via
-    #   -r requirements.txt
+    #   -r ./requirements.txt
     #   diffusers
 adam-mini==1.0.1
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 aiohappyeyeballs==2.4.0
     # via aiohttp
 aiohttp==3.10.5
     # via fsspec
 aiosignal==1.3.1
     # via aiohttp
 altair==4.2.2
-    # via -r requirements.txt
-annotated-types==0.7.0
-    # via pydantic
-anyio==4.4.0
-    # via starlette
+    # via -r ./requirements.txt
 astunparse==1.6.3
     # via tensorflow-intel
 async-timeout==4.0.3
@@ -34,41 +30,34 @@ attrs==24.2.0
     #   aiohttp
     #   jsonschema
     #   referencing
-bitsandbytes==0.43.3
-    # via -r requirements.txt
+bitsandbytes==0.44.0
+    # via -r ./requirements.txt
 certifi==2022.12.7
     # via
     #   requests
     #   sentry-sdk
 charset-normalizer==2.1.1
     # via requests
 click==8.1.7
-    # via
-    #   uvicorn
-    #   wandb
+    # via wandb
 colorama==0.4.6
     # via
     #   click
     #   tqdm
 coloredlogs==15.0.1
     # via onnxruntime-gpu
 dadaptation==3.2
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 diffusers==0.25.0
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 docker-pycreds==0.4.0
     # via wandb
 easygui==0.98.3
-    # via -r requirements.txt
-einops==0.7.0
-    # via
-    #   -r requirements.txt
+    # via -r ./requirements.txt
+einops==0.8.0
+    # via -r ./requirements.txt
 entrypoints==0.4
     # via altair
-exceptiongroup==1.2.2
-    # via anyio
-fastapi==0.114.0
-    # via -r requirements.txt
 filelock==3.13.1
     # via
     #   diffusers
@@ -89,7 +78,7 @@ fsspec==2024.2.0
     #   pytorch-lightning
     #   torch
 ftfy==6.1.1
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 gast==0.6.0
     # via tensorflow-intel
 gitdb==4.0.11
@@ -102,15 +91,13 @@ grpcio==1.66.1
     # via
     #   tensorboard
     #   tensorflow-intel
-h11==0.14.0
-    # via uvicorn
 h5py==3.11.0
     # via
     #   keras
     #   tensorflow-intel
 huggingface-hub==0.24.5
     # via
-    #   -r requirements.txt
+    #   -r ./requirements.txt
     #   accelerate
     #   diffusers
     #   timm
@@ -120,11 +107,10 @@ humanfriendly==10.0
     # via coloredlogs
 idna==3.4
     # via
-    #   anyio
     #   requests
     #   yarl
 imagesize==1.4.1
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 importlib-metadata==8.4.0
     # via diffusers
 jinja2==3.1.3
@@ -144,7 +130,7 @@ lightning-utilities==0.11.7
     #   pytorch-lightning
     #   torchmetrics
 lion-pytorch==0.0.6
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 markdown==3.7
     # via tensorboard
 markdown-it-py==3.0.0
@@ -184,19 +170,18 @@ numpy==1.26.3
     #   opt-einsum
     #   pandas
     #   pytorch-lightning
-    #   pytorch-optimizer
     #   scipy
     #   tensorboard
     #   tensorflow-intel
     #   torchmetrics
     #   torchvision
     #   transformers
 onnx==1.15.0
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 onnxruntime-gpu==1.19.0
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 opencv-python==4.8.1.78
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 opt-einsum==3.3.0
     # via tensorflow-intel
 optree==0.12.1
@@ -223,10 +208,10 @@ pillow==10.2.0
 platformdirs==4.3.2
     # via wandb
 prodigyopt==1.0
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 protobuf==3.20.3
     # via
-    #   -r requirements.txt
+    #   -r ./requirements.txt
     #   onnx
     #   onnxruntime-gpu
     #   tensorboard
@@ -236,20 +221,14 @@ psutil==6.0.0
     # via
     #   accelerate
     #   wandb
-pydantic==2.9.1
-    # via fastapi
-pydantic-core==2.23.3
-    # via pydantic
 pygments==2.18.0
     # via rich
 pyreadline3==3.4.1
     # via humanfriendly
 python-dateutil==2.9.0.post0
     # via pandas
 pytorch-lightning==1.9.0
-    # via -r requirements.txt
-pytorch-optimizer==3.1.2
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 pytz==2024.2
     # via pandas
 pyyaml==6.0.2
@@ -277,25 +256,25 @@ requests==2.28.1
     #   wandb
 rich==13.7.0
     # via
-    #   -r requirements.txt
+    #   -r ./requirements.txt
     #   keras
 rpds-py==0.20.0
     # via
     #   jsonschema
     #   referencing
 safetensors==0.4.4
     # via
-    #   -r requirements.txt
+    #   -r ./requirements.txt
     #   accelerate
     #   diffusers
     #   timm
     #   transformers
 schedulefree==1.2.7
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 scipy==1.14.1
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 sentencepiece==0.2.0
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 sentry-sdk==2.14.0
     # via wandb
 setproctitle==1.3.3
@@ -316,89 +295,73 @@ six==1.16.0
     #   tensorflow-intel
 smmap==5.0.1
     # via gitdb
-sniffio==1.3.1
-    # via anyio
-starlette==0.38.5
-    # via fastapi
 sympy==1.12
     # via
     #   onnxruntime-gpu
     #   torch
 tensorboard==2.17.1
     # via
-    #   -r requirements.txt
+    #   -r ./requirements.txt
     #   tensorflow-intel
 tensorboard-data-server==0.7.2
     # via tensorboard
 tensorflow==2.17.0
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 tensorflow-io-gcs-filesystem==0.31.0
     # via tensorflow-intel
 termcolor==2.4.0
     # via tensorflow-intel
 timm==1.0.7
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 tokenizers==0.19.1
     # via transformers
 toml==0.10.2
-    # via
-    #   -r requirements.txt
+    # via -r ./requirements.txt
 toolz==0.12.1
     # via altair
 torch==2.4.1+cu124
     # via
-    #   -r requirements.txt
     #   accelerate
     #   bitsandbytes
     #   diffusers
     #   lion-pytorch
     #   pytorch-lightning
-    #   pytorch-optimizer
     #   timm
     #   torch-optimi
     #   torchmetrics
     #   torchvision
 torch-optimi==0.2.1
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 torchmetrics==1.0.3
     # via pytorch-lightning
 torchvision==0.19.1+cu124
-    # via
-    #   -r requirements.txt
-    #   timm
+    # via timm
 tqdm==4.64.1
     # via
     #   huggingface-hub
     #   pytorch-lightning
     #   transformers
 transformers==4.44.2
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 typing-extensions==4.9.0
     # via
-    #   anyio
-    #   fastapi
     #   huggingface-hub
     #   lightning-utilities
     #   multidict
     #   optree
-    #   pydantic
-    #   pydantic-core
     #   pytorch-lightning
     #   tensorflow-intel
     #   torch
-    #   uvicorn
 tzdata==2024.1
     # via pandas
 urllib3==1.26.13
     # via
     #   requests
     #   sentry-sdk
-uvicorn==0.30.6
-    # via -r requirements.txt
 voluptuous==0.13.1
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 wandb==0.17.9
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 wcwidth==0.2.13
     # via ftfy
 werkzeug==3.0.4

diff --git a/requirements.txt b/requirements.txt
@@ -4,11 +4,14 @@ diffusers[torch]==0.25.0
 ftfy==6.1.1
 # albumentations==1.3.0
 opencv-python==4.8.1.78
-einops==0.7.0
+einops==0.8.0
 pytorch-lightning==1.9.0
 bitsandbytes==0.44.0
-prodigyopt==1.0
-lion-pytorch==0.0.6
+prodigyopt
+lion-pytorch
+scipy
+dadaptation
+wandb
 schedulefree==1.2.7
 torch-optimi==0.2.1
 adam-mini==1.0.1
@@ -40,7 +43,7 @@ onnxruntime-gpu>=1.18.1
 timm==1.0.7
 
 # this is for onnx: 
-protobuf==3.20.3
+protobuf
 # open clip for SDXL
 # open-clip-torch==2.20.0
 # For logging