update

sdbds · Feb 17, 2025 · 3744a2c · 3744a2c
1 parent 83358fa
commit 3744a2c
Show file tree

Hide file tree

Showing 2 changed files with 47 additions and 38 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 --index-strategy unsafe-best-match --no-build-isolation -p 3.10
 -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.1.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
+    # via -r ./requirements.txt
 astunparse==1.6.3
     # via tensorflow-intel
 async-timeout==4.0.3
@@ -31,7 +31,7 @@ attrs==24.2.0
     #   jsonschema
     #   referencing
 bitsandbytes==0.44.1
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 certifi==2022.12.7
     # via
     #   requests
@@ -47,15 +47,17 @@ 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
+    #   flash-attn
 entrypoints==0.4
     # via altair
 filelock==3.13.1
@@ -64,6 +66,8 @@ filelock==3.13.1
     #   huggingface-hub
     #   torch
     #   transformers
+flash-attn @ https://github.com/kingbri1/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu124torch2.5.1cxx11abiFALSE-cp310-cp310-win_amd64.whl
+    # via -r ./requirements.txt
 flatbuffers==24.3.25
     # via
     #   onnxruntime-gpu
@@ -78,7 +82,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
@@ -96,10 +100,10 @@ h5py==3.11.0
     #   keras
     #   tensorflow-intel
 heavyball==0.24.4
-    # via -r requirements.txt
+    # via -r ./requirements.txt
 huggingface-hub==0.24.5
     # via
-    #   -r requirements.txt
+    #   -r ./requirements.txt
     #   accelerate
     #   diffusers
     #   timm
@@ -112,7 +116,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 +136,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
@@ -180,11 +184,11 @@ 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
     #   heavyball
@@ -211,16 +215,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
+    # 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
@@ -237,7 +241,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
@@ -265,25 +269,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.4
-    # 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
@@ -310,32 +314,33 @@ 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
-#tensorflow-intel==2.17.0
+    # 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
+    #   flash-attn
     #   heavyball
     #   lion-pytorch
     #   prodigy-plus-schedule-free
@@ -345,7 +350,7 @@ torch==2.5.1+cu124
     #   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
@@ -356,7 +361,9 @@ tqdm==4.64.1
     #   pytorch-lightning
     #   transformers
 transformers==4.44.2
-    # via -r requirements.txt
+    # via -r ./requirements.txt
+triton @ https://github.com/woct0rdho/triton-windows/releases/download/v3.2.0-windows.post9/triton-3.2.0-cp310-cp310-win_amd64.whl
+    # via -r ./requirements.txt
 typing-extensions==4.9.0
     # via
     #   huggingface-hub
@@ -373,9 +380,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
@@ -56,3 +56,5 @@ rich==13.7.0
 sentencepiece==0.2.0
 # for kohya_ss library
 -e .
+https://github.com/woct0rdho/triton-windows/releases/download/v3.2.0-windows.post9/triton-3.2.0-cp310-cp310-win_amd64.whl
+https://github.com/kingbri1/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu124torch2.5.1cxx11abiFALSE-cp310-cp310-win_amd64.whl