update requirements

sdbds · Dec 4, 2024 · af36dc2 · af36dc2
1 parent d29b4fb
commit af36dc2
Show file tree

Hide file tree

Showing 2 changed files with 53 additions and 45 deletions.
diff --git a/requirements-uv.txt b/requirements-uv.txt
@@ -1,26 +1,26 @@
 # This file was autogenerated by uv via the following command:
-#    uv pip compile ./requirements.txt -o requirements-uv.txt -i https://pypi.org/simple --index-strategy unsafe-best-match
+#    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
+adam-mini==1.1.1
+    # 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
+    # via -r requirements.txt
 astunparse==1.6.3
     # via tensorflow-intel
 async-timeout==4.0.3
@@ -30,8 +30,8 @@ attrs==24.2.0
     #   aiohttp
     #   jsonschema
     #   referencing
-bitsandbytes==0.44.0
-    # via -r ./requirements.txt
+bitsandbytes==0.44.1
+    # via -r requirements.txt
 certifi==2022.12.7
     # via
     #   requests
@@ -47,15 +47,15 @@ colorama==0.4.6
 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
+    # via -r requirements.txt
 einops==0.8.0
-    # via -r ./requirements.txt
+    # via -r requirements.txt
 entrypoints==0.4
     # via altair
 filelock==3.13.1
@@ -78,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
@@ -95,11 +95,11 @@ h5py==3.11.0
     # via
     #   keras
     #   tensorflow-intel
-heavyball==0.10.2
-    # via -r ./requirements.txt
+heavyball==0.24.4
+    # via -r requirements.txt
 huggingface-hub==0.24.5
     # via
-    #   -r ./requirements.txt
+    #   -r requirements.txt
     #   accelerate
     #   diffusers
     #   timm
@@ -112,7 +112,7 @@ idna==3.4
     #   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
@@ -132,7 +132,7 @@ lightning-utilities==0.11.8
     #   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
@@ -164,6 +164,7 @@ numpy==1.26.3
     #   bitsandbytes
     #   diffusers
     #   h5py
+    #   heavyball
     #   keras
     #   ml-dtypes
     #   onnx
@@ -179,13 +180,15 @@ numpy==1.26.3
     #   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
+    # via
+    #   heavyball
+    #   tensorflow-intel
 optree==0.12.1
     # via keras
 packaging==22.0
@@ -208,14 +211,16 @@ pillow==10.2.0
     #   diffusers
     #   torchvision
 pillow-avif-plugin==1.4.6
-    # via -r ./requirements.txt
+    # via -r requirements.txt
 platformdirs==4.3.2
     # via wandb
+prodigy-plus-schedule-free==1.7.0
+    # via -r requirements.txt
 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
@@ -232,7 +237,7 @@ pyreadline3==3.4.1
 python-dateutil==2.9.0.post0
     # via pandas
 pytorch-lightning==1.9.0
-    # via -r ./requirements.txt
+    # via -r requirements.txt
 pytz==2024.2
     # via pandas
 pyyaml==6.0.2
@@ -260,25 +265,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.3
-    # via -r ./requirements.txt
+schedulefree==1.4
+    # 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
@@ -305,40 +310,42 @@ sympy==1.13.1
     #   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-intel==2.17.0
     # via tensorflow
 tensorflow-io-gcs-filesystem==0.31.0
     # via tensorflow-intel
 termcolor==2.4.0
     # via tensorflow-intel
 timm==1.0.11
-    # 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.5.1+cu124
     # via
-    #   -r ./requirements.txt
+    #   -r requirements.txt
     #   accelerate
     #   bitsandbytes
     #   diffusers
+    #   heavyball
     #   lion-pytorch
+    #   prodigy-plus-schedule-free
     #   pytorch-lightning
     #   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.20.1+cu124
@@ -349,7 +356,7 @@ tqdm==4.64.1
     #   pytorch-lightning
     #   transformers
 transformers==4.44.2
-    # via -r ./requirements.txt
+    # via -r requirements.txt
 typing-extensions==4.9.0
     # via
     #   huggingface-hub
@@ -366,9 +373,9 @@ urllib3==1.26.13
     #   requests
     #   sentry-sdk
 voluptuous==0.13.1
-    # via -r ./requirements.txt
+    # via -r requirements.txt
 wandb==0.18.0
-    # 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
@@ -6,17 +6,18 @@ ftfy==6.1.1
 opencv-python==4.8.1.78
 einops==0.8.0
 pytorch-lightning==1.9.0
-bitsandbytes==0.44.0
+bitsandbytes==0.44.1
 prodigyopt
+prodigy-plus-schedule-free
 lion-pytorch
 scipy
 dadaptation
-heavyball>=0.10.2
+heavyball>=0.24.4
 wandb==0.18.0
 pillow-avif-plugin
-schedulefree==1.3.0
+schedulefree>=1.4.0
 torch-optimi==0.2.1
-adam-mini==1.0.1
+adam-mini>=1.1.1
 tensorboard
 safetensors==0.4.4
 # gradio==3.16.2