diff --git a/requirements-uv.txt b/requirements-uv.txt index 6dd48a0ab..23c46e31b 100644 --- a/requirements-uv.txt +++ b/requirements-uv.txt @@ -1,7 +1,7 @@ # 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 @@ -9,10 +9,10 @@ absl-py==2.1.0 # 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 @@ -20,7 +20,7 @@ aiohttp==3.10.5 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,7 +269,7 @@ requests==2.28.1 # wandb rich==13.7.0 # via - # -r requirements.txt + # -r ./requirements.txt # keras rpds-py==0.20.0 # via @@ -273,17 +277,17 @@ rpds-py==0.20.0 # 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 index 4f1cdb1d1..49c366fed 100644 --- 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