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: true
#   with-sources: false
#   generate-hashes: false
#   universal: false

-e file:.
aiohappyeyeballs==2.4.2
    # via aiohttp
aiohttp==3.10.6
    # via datasets
    # via fsspec
aiosignal==1.3.1
    # via aiohttp
annotated-types==0.7.0
    # via pydantic
arrow==1.3.0
    # via isoduration
async-timeout==4.0.3
    # via aiohttp
attrs==24.2.0
    # via aiohttp
    # via jsonschema
    # via referencing
babel==2.16.0
    # via mkdocs-material
bitsandbytes==0.44.0
    # via refiners
black==24.8.0
    # via refiners
boto3==1.35.28
    # via neptune
botocore==1.35.28
    # via boto3
    # via s3transfer
bravado==11.0.3
    # via neptune
bravado-core==6.1.1
    # via bravado
certifi==2024.8.30
    # via requests
    # via sentry-sdk
charset-normalizer==3.3.2
    # via requests
click==8.1.7
    # via black
    # via mkdocs
    # via mkdocstrings
    # via neptune
    # via wandb
colorama==0.4.6
    # via griffe
    # via mkdocs-material
datasets==3.0.1
    # via refiners
diffusers==0.30.3
    # via refiners
dill==0.3.8
    # via datasets
    # via multiprocess
docker-pycreds==0.4.0
    # via wandb
exceptiongroup==1.2.2
    # via pytest
filelock==3.16.1
    # via datasets
    # via diffusers
    # via huggingface-hub
    # via torch
    # via transformers
    # via triton
fqdn==1.5.1
    # via jsonschema
frozenlist==1.4.1
    # via aiohttp
    # via aiosignal
fsspec==2024.6.1
    # via datasets
    # via huggingface-hub
    # via torch
future==1.0.0
    # via neptune
ghp-import==2.1.0
    # via mkdocs
gitdb==4.0.11
    # via gitpython
gitpython==3.1.43
    # via neptune
    # via refiners
    # via wandb
griffe==1.3.1
    # via mkdocstrings-python
huggingface-hub==0.25.1
    # via datasets
    # via diffusers
    # via refiners
    # via timm
    # via tokenizers
    # via transformers
idna==3.10
    # via jsonschema
    # via requests
    # via yarl
importlib-metadata==8.5.0
    # via diffusers
importlib-resources==6.4.5
    # via swagger-spec-validator
iniconfig==2.0.0
    # via pytest
isoduration==20.11.0
    # via jsonschema
jaxtyping==0.2.34
    # via refiners
jinja2==3.1.4
    # via mkdocs
    # via mkdocs-material
    # via mkdocstrings
    # via torch
jmespath==1.0.1
    # via boto3
    # via botocore
jsonpointer==3.0.0
    # via jsonschema
jsonref==1.1.0
    # via bravado-core
jsonschema==4.23.0
    # via bravado-core
    # via swagger-spec-validator
jsonschema-specifications==2023.12.1
    # via jsonschema
loguru==0.7.2
    # via refiners
markdown==3.7
    # via mkdocs
    # via mkdocs-autorefs
    # via mkdocs-material
    # via mkdocstrings
    # via pymdown-extensions
markupsafe==2.1.5
    # via jinja2
    # via mkdocs
    # via mkdocs-autorefs
    # via mkdocstrings
mergedeep==1.3.4
    # via mkdocs
    # via mkdocs-get-deps
mkdocs==1.6.1
    # via mkdocs-autorefs
    # via mkdocs-literate-nav
    # via mkdocs-material
    # via mkdocstrings
mkdocs-autorefs==1.2.0
    # via mkdocstrings
    # via mkdocstrings-python
mkdocs-get-deps==0.2.0
    # via mkdocs
mkdocs-literate-nav==0.6.1
    # via refiners
mkdocs-material==9.5.38
    # via refiners
mkdocs-material-extensions==1.3.1
    # via mkdocs-material
mkdocstrings==0.26.1
    # via mkdocstrings-python
    # via refiners
mkdocstrings-python==1.11.1
    # via mkdocstrings
monotonic==1.6
    # via bravado
mpmath==1.3.0
    # via sympy
msgpack==1.1.0
    # via bravado
    # via bravado-core
multidict==6.1.0
    # via aiohttp
    # via yarl
multiprocess==0.70.16
    # via datasets
mypy-extensions==1.0.0
    # via black
neptune==1.11.1
    # via refiners
networkx==3.3
    # via torch
numpy==2.1.1
    # via bitsandbytes
    # via datasets
    # via diffusers
    # via pandas
    # via pyarrow
    # via refiners
    # via torchvision
    # via transformers
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==9.1.0.70
    # 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.6.68
    # via nvidia-cusolver-cu12
    # via nvidia-cusparse-cu12
nvidia-nvtx-cu12==12.1.105
    # via torch
oauthlib==3.2.2
    # via neptune
    # via requests-oauthlib
packaging==24.1
    # via black
    # via datasets
    # via huggingface-hub
    # via mkdocs
    # via neptune
    # via pytest
    # via pytest-rerunfailures
    # via refiners
    # via transformers
paginate==0.5.7
    # via mkdocs-material
pandas==2.2.3
    # via datasets
    # via neptune
pathspec==0.12.1
    # via black
    # via mkdocs
pillow==10.4.0
    # via diffusers
    # via neptune
    # via refiners
    # via torchvision
piq==0.8.0
    # via refiners
platformdirs==4.3.6
    # via black
    # via mkdocs-get-deps
    # via mkdocstrings
    # via wandb
pluggy==1.5.0
    # via pytest
prodigyopt==1.0
    # via refiners
protobuf==5.28.2
    # via wandb
psutil==6.0.0
    # via neptune
    # via wandb
pyarrow==17.0.0
    # via datasets
pydantic==2.9.2
    # via refiners
pydantic-core==2.23.4
    # via pydantic
pygments==2.18.0
    # via mkdocs-material
pyjwt==2.9.0
    # via neptune
pymdown-extensions==10.10.2
    # via mkdocs-material
    # via mkdocstrings
pytest==8.3.3
    # via pytest-rerunfailures
pytest-rerunfailures==14.0
    # via refiners
python-dateutil==2.9.0.post0
    # via arrow
    # via botocore
    # via bravado
    # via bravado-core
    # via ghp-import
    # via pandas
pytz==2024.2
    # via bravado-core
    # via pandas
pyyaml==6.0.2
    # via bravado
    # via bravado-core
    # via datasets
    # via huggingface-hub
    # via mkdocs
    # via mkdocs-get-deps
    # via pymdown-extensions
    # via pyyaml-env-tag
    # via swagger-spec-validator
    # via timm
    # via transformers
    # via wandb
pyyaml-env-tag==0.1
    # via mkdocs
referencing==0.35.1
    # via jsonschema
    # via jsonschema-specifications
regex==2024.9.11
    # via diffusers
    # via mkdocs-material
    # via transformers
requests==2.32.3
    # via bravado
    # via bravado-core
    # via datasets
    # via diffusers
    # via huggingface-hub
    # via mkdocs-material
    # via neptune
    # via refiners
    # via requests-oauthlib
    # via transformers
    # via wandb
requests-oauthlib==2.0.0
    # via neptune
rfc3339-validator==0.1.4
    # via jsonschema
rfc3986-validator==0.1.1
    # via jsonschema
rpds-py==0.20.0
    # via jsonschema
    # via referencing
s3transfer==0.10.2
    # via boto3
safetensors==0.4.5
    # via diffusers
    # via refiners
    # via timm
    # via transformers
segment-anything-hq==0.3
    # via refiners
segment-anything-py==1.0.1
    # via refiners
sentencepiece==0.2.0
    # via refiners
sentry-sdk==2.14.0
    # via wandb
setproctitle==1.3.3
    # via wandb
setuptools==75.1.0
    # via wandb
simplejson==3.19.3
    # via bravado
    # via bravado-core
six==1.16.0
    # via bravado
    # via bravado-core
    # via docker-pycreds
    # via neptune
    # via python-dateutil
    # via rfc3339-validator
smmap==5.0.1
    # via gitdb
swagger-spec-validator==3.0.4
    # via bravado-core
    # via neptune
sympy==1.13.3
    # via torch
timm==1.0.9
    # via refiners
tokenizers==0.20.0
    # via transformers
tomli==2.0.1
    # via black
    # via pytest
    # via refiners
torch==2.4.1
    # via bitsandbytes
    # via refiners
    # via segment-anything-hq
    # via segment-anything-py
    # via timm
    # via torchvision
torchvision==0.19.1
    # via piq
    # via refiners
    # via segment-anything-hq
    # via segment-anything-py
    # via timm
tqdm==4.66.5
    # via datasets
    # via huggingface-hub
    # via refiners
    # via transformers
transformers==4.45.1
    # via refiners
triton==3.0.0
    # via torch
typeguard==2.13.3
    # via jaxtyping
types-python-dateutil==2.9.0.20240906
    # via arrow
typing-extensions==4.12.2
    # via black
    # via bravado
    # via huggingface-hub
    # via multidict
    # via neptune
    # via pydantic
    # via pydantic-core
    # via swagger-spec-validator
    # via torch
tzdata==2024.2
    # via pandas
uri-template==1.3.0
    # via jsonschema
urllib3==2.2.3
    # via botocore
    # via neptune
    # via requests
    # via sentry-sdk
wandb==0.18.1
    # via refiners
watchdog==5.0.2
    # via mkdocs
webcolors==24.8.0
    # via jsonschema
websocket-client==1.8.0
    # via neptune
xxhash==3.5.0
    # via datasets
yarl==1.13.0
    # via aiohttp
zipp==3.20.2
    # via importlib-metadata