requirements.lock

# generated by rye
# use `rye lock` or `rye sync` to update this lockfile
#
# last locked with the following flags:
#   pre: false
#   features: []
#   all-features: false
#   with-sources: false
#   generate-hashes: false
#   universal: false

-e file:.
absl-py==2.1.0
    # via tensorboard
accelerate==0.30.0
    # via lightning-diffusion
    # via peft
aiohttp==3.9.5
    # via datasets
    # via fsspec
aiosignal==1.3.1
    # via aiohttp
annotated-types==0.7.0
    # via pydantic
antlr4-python3-runtime==4.9.3
    # via hydra-core
    # via omegaconf
anyio==4.4.0
    # via httpx
    # via openai
attrs==23.2.0
    # via aiohttp
autogen==0.3.0
    # via lightning-diffusion
beautifulsoup4==4.12.3
    # via bs4
bitsandbytes==0.41.0
    # via lightning
braceexpand==0.1.7
    # via webdataset
bs4==0.0.2
    # via lightning-diffusion
certifi==2024.2.2
    # via httpcore
    # via httpx
    # via requests
    # via sentry-sdk
charset-normalizer==3.3.2
    # via requests
click==8.1.7
    # via wandb
contourpy==1.2.1
    # via matplotlib
cycler==0.12.1
    # via matplotlib
datasets==2.19.0
    # via lightning-diffusion
decord==0.6.0
    # via lightning-diffusion
deepspeed==0.15.1
    # via lightning-diffusion
diffusers @ git+https://github.com/huggingface/diffusers@7f323f0f3190533e596e09a4923dad1f73f23a91
    # via lightning-diffusion
dill==0.3.8
    # via datasets
    # via multiprocess
diskcache==5.6.3
    # via autogen
distro==1.9.0
    # via openai
docker==7.1.0
    # via autogen
docker-pycreds==0.4.0
    # via wandb
docstring-parser==0.16
    # via jsonargparse
einops==0.8.0
    # via lightning-diffusion
filelock==3.14.0
    # via datasets
    # via diffusers
    # via huggingface-hub
    # via torch
    # via transformers
    # via triton
flaml==2.2.0
    # via autogen
fonttools==4.51.0
    # via matplotlib
frozenlist==1.4.1
    # via aiohttp
    # via aiosignal
fsspec==2024.3.1
    # via datasets
    # via huggingface-hub
    # via lightning
    # via pytorch-lightning
    # via torch
ftfy==6.2.0
    # via lightning-diffusion
gitdb==4.0.11
    # via gitpython
gitpython==3.1.43
    # via wandb
grpcio==1.63.0
    # via tensorboard
h11==0.14.0
    # via httpcore
hjson==3.1.0
    # via deepspeed
httpcore==1.0.5
    # via httpx
httpx==0.27.2
    # via openai
huggingface-hub==0.23.3
    # via accelerate
    # via datasets
    # via diffusers
    # via peft
    # via tokenizers
    # via transformers
hydra-core==1.3.2
    # via lightning
idna==3.7
    # via anyio
    # via httpx
    # via requests
    # via yarl
importlib-metadata==7.1.0
    # via diffusers
importlib-resources==6.4.0
    # via typeshed-client
jinja2==3.1.3
    # via torch
jiter==0.5.0
    # via openai
jsonargparse==4.28.0
    # via lightning
kiwisolver==1.4.5
    # via matplotlib
lightning==2.2.4
    # via lightning-diffusion
lightning-utilities==0.11.2
    # via lightning
    # via pytorch-lightning
    # via torchmetrics
loguru==0.7.2
    # via lightning-diffusion
markdown==3.6
    # via tensorboard
markdown-it-py==3.0.0
    # via rich
markupsafe==2.1.5
    # via jinja2
    # via werkzeug
matplotlib==3.8.4
    # via lightning
mdurl==0.1.2
    # via markdown-it-py
mpmath==1.3.0
    # via sympy
multidict==6.0.5
    # via aiohttp
    # via yarl
multiprocess==0.70.16
    # via datasets
networkx==3.3
    # via torch
ninja==1.11.1.1
    # via deepspeed
numpy==1.26.4
    # via accelerate
    # via autogen
    # via contourpy
    # via datasets
    # via decord
    # via deepspeed
    # via diffusers
    # via flaml
    # via lightning
    # via matplotlib
    # via opencv-python
    # via pandas
    # via peft
    # via pyarrow
    # via pytorch-lightning
    # via scipy
    # via tensorboard
    # via tensorboardx
    # via torchmetrics
    # via torchvision
    # via transformers
    # via webdataset
nvidia-cublas-cu12==12.1.3.1
    # via nvidia-cudnn-cu12
    # via nvidia-cusolver-cu12
    # via torch
nvidia-cuda-cupti-cu12==12.1.105
    # via torch
nvidia-cuda-nvrtc-cu12==12.1.105
    # via torch
nvidia-cuda-runtime-cu12==12.1.105
    # via torch
nvidia-cudnn-cu12==8.9.2.26
    # via torch
nvidia-cufft-cu12==11.0.2.54
    # via torch
nvidia-curand-cu12==10.3.2.106
    # via torch
nvidia-cusolver-cu12==11.4.5.107
    # via torch
nvidia-cusparse-cu12==12.1.0.106
    # via nvidia-cusolver-cu12
    # via torch
nvidia-nccl-cu12==2.20.5
    # via torch
nvidia-nvjitlink-cu12==12.4.127
    # via nvidia-cusolver-cu12
    # via nvidia-cusparse-cu12
nvidia-nvtx-cu12==12.1.105
    # via torch
omegaconf==2.3.0
    # via hydra-core
    # via lightning
    # via lightning-diffusion
openai==1.45.1
    # via autogen
opencv-python==4.9.0.80
    # via lightning-diffusion
packaging==24.0
    # via accelerate
    # via autogen
    # via datasets
    # via deepspeed
    # via huggingface-hub
    # via hydra-core
    # via lightning
    # via lightning-utilities
    # via matplotlib
    # via peft
    # via pytorch-lightning
    # via tensorboardx
    # via torchmetrics
    # via transformers
pandas==2.2.2
    # via datasets
peft==0.10.0
    # via lightning-diffusion
pillow==10.3.0
    # via diffusers
    # via matplotlib
    # via torchvision
platformdirs==4.2.2
    # via wandb
protobuf==5.26.1
    # via tensorboard
    # via tensorboardx
    # via wandb
psutil==5.9.8
    # via accelerate
    # via deepspeed
    # via peft
    # via wandb
py-cpuinfo==9.0.0
    # via deepspeed
pyarrow==16.0.0
    # via datasets
pyarrow-hotfix==0.6
    # via datasets
pydantic==2.9.1
    # via autogen
    # via deepspeed
    # via openai
pydantic-core==2.23.3
    # via pydantic
pygments==2.17.2
    # via rich
pyparsing==3.1.2
    # via matplotlib
python-dateutil==2.9.0.post0
    # via matplotlib
    # via pandas
python-dotenv==1.0.1
    # via autogen
pytorch-lightning==2.2.4
    # via lightning
pytz==2024.1
    # via pandas
pyyaml==6.0.1
    # via accelerate
    # via datasets
    # via huggingface-hub
    # via jsonargparse
    # via lightning
    # via omegaconf
    # via peft
    # via pytorch-lightning
    # via transformers
    # via wandb
    # via webdataset
regex==2024.4.28
    # via diffusers
    # via tiktoken
    # via transformers
requests==2.31.0
    # via datasets
    # via diffusers
    # via docker
    # via huggingface-hub
    # via tiktoken
    # via transformers
    # via wandb
rich==13.7.1
    # via lightning
safetensors==0.4.3
    # via accelerate
    # via diffusers
    # via peft
    # via transformers
scipy==1.13.0
    # via lightning-diffusion
sentencepiece==0.2.0
    # via lightning-diffusion
sentry-sdk==2.5.1
    # via wandb
setproctitle==1.3.3
    # via wandb
setuptools==69.5.1
    # via lightning-utilities
    # via tensorboard
    # via wandb
six==1.16.0
    # via docker-pycreds
    # via python-dateutil
    # via tensorboard
smmap==5.0.1
    # via gitdb
sniffio==1.3.1
    # via anyio
    # via httpx
    # via openai
soupsieve==2.5
    # via beautifulsoup4
sympy==1.12
    # via torch
tensorboard==2.16.2
    # via lightning-diffusion
tensorboard-data-server==0.7.2
    # via tensorboard
tensorboardx==2.6.2.2
    # via lightning
termcolor==2.4.0
    # via autogen
tiktoken==0.7.0
    # via autogen
tokenizers==0.19.1
    # via transformers
torch==2.3.0
    # via accelerate
    # via deepspeed
    # via lightning
    # via lightning-diffusion
    # via peft
    # via pytorch-lightning
    # via torchmetrics
    # via torchvision
torchmetrics==1.3.2
    # via lightning
    # via pytorch-lightning
torchvision==0.18.0
    # via lightning-diffusion
tqdm==4.66.2
    # via datasets
    # via deepspeed
    # via huggingface-hub
    # via lightning
    # via openai
    # via peft
    # via pytorch-lightning
    # via transformers
transformers==4.40.1
    # via lightning-diffusion
    # via peft
triton==2.3.0
    # via torch
typeshed-client==2.5.1
    # via jsonargparse
typing-extensions==4.11.0
    # via huggingface-hub
    # via lightning
    # via lightning-utilities
    # via openai
    # via pydantic
    # via pydantic-core
    # via pytorch-lightning
    # via torch
    # via typeshed-client
tzdata==2024.1
    # via pandas
urllib3==2.2.1
    # via docker
    # via requests
    # via sentry-sdk
wandb==0.17.1
    # via lightning-diffusion
wcwidth==0.2.13
    # via ftfy
webdataset==0.2.86
    # via lightning-diffusion
werkzeug==3.0.3
    # via tensorboard
xxhash==3.4.1
    # via datasets
yarl==1.9.4
    # via aiohttp
zipp==3.18.1
    # via importlib-metadata