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.4
    # via aiohttp
aiohttp==3.11.11
    # via datasets
    # via fsspec
aiosignal==1.3.2
    # via aiohttp
annotated-types==0.7.0
    # via pydantic
arrow==1.3.0
    # via isoduration
async-timeout==5.0.1
    # via aiohttp
attrs==24.3.0
    # via aiohttp
    # via jsonschema
    # via referencing
babel==2.16.0
    # via mkdocs-material
bitsandbytes==0.45.0
    # via refiners
black==24.10.0
    # via refiners
boto3==1.35.84
    # via neptune
botocore==1.35.84
    # via boto3
    # via s3transfer
bravado==11.0.3
    # via neptune
bravado-core==6.1.1
    # via bravado
certifi==2024.12.14
    # via requests
    # via sentry-sdk
charset-normalizer==3.4.0
    # 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.2.0
    # via refiners
diffusers==0.31.0
    # 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.5.0
    # via aiohttp
    # via aiosignal
fsspec==2024.9.0
    # 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.5.1
    # via mkdocstrings-python
huggingface-hub==0.27.0
    # 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.36
    # 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==2024.10.1
    # via jsonschema
loguru==0.7.3
    # via refiners
markdown==3.7
    # via mkdocs
    # via mkdocs-autorefs
    # via mkdocs-material
    # via mkdocstrings
    # via pymdown-extensions
markupsafe==3.0.2
    # 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.49
    # via refiners
mkdocs-material-extensions==1.3.1
    # via mkdocs-material
mkdocstrings==0.27.0
    # via mkdocstrings-python
    # via refiners
mkdocstrings-python==1.12.2
    # 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.13.0
    # via refiners
networkx==3.4.2
    # via torch
numpy==2.2.0
    # via bitsandbytes
    # via datasets
    # via diffusers
    # via pandas
    # via refiners
    # via torchvision
    # via transformers
nvidia-cublas-cu12==12.4.5.8
    # via nvidia-cudnn-cu12
    # via nvidia-cusolver-cu12
    # via torch
nvidia-cuda-cupti-cu12==12.4.127
    # via torch
nvidia-cuda-nvrtc-cu12==12.4.127
    # via torch
nvidia-cuda-runtime-cu12==12.4.127
    # via torch
nvidia-cudnn-cu12==9.1.0.70
    # via torch
nvidia-cufft-cu12==11.2.1.3
    # via torch
nvidia-curand-cu12==10.3.5.147
    # via torch
nvidia-cusolver-cu12==11.6.1.9
    # via torch
nvidia-cusparse-cu12==12.3.1.170
    # via nvidia-cusolver-cu12
    # via torch
nvidia-nccl-cu12==2.21.5
    # via torch
nvidia-nvjitlink-cu12==12.4.127
    # via nvidia-cusolver-cu12
    # via nvidia-cusparse-cu12
    # via torch
nvidia-nvtx-cu12==12.4.127
    # via torch
oauthlib==3.2.2
    # via neptune
    # via requests-oauthlib
packaging==24.2
    # 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==11.0.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.1.1
    # via refiners
propcache==0.2.1
    # via aiohttp
    # via yarl
protobuf==5.29.2
    # via wandb
psutil==6.1.0
    # via neptune
    # via wandb
pyarrow==18.1.0
    # via datasets
pydantic==2.10.4
    # via refiners
    # via wandb
pydantic-core==2.27.2
    # via pydantic
pygments==2.18.0
    # via mkdocs-material
pyjwt==2.10.1
    # via neptune
pymdown-extensions==10.12
    # via mkdocs-material
    # via mkdocstrings
pytest==8.3.4
    # via pytest-rerunfailures
pytest-rerunfailures==15.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.11.6
    # 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.22.3
    # via jsonschema
    # via referencing
s3transfer==0.10.4
    # 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.19.2
    # via wandb
setproctitle==1.3.4
    # via wandb
setuptools==75.6.0
    # via wandb
simplejson==3.19.3
    # via bravado
    # via bravado-core
six==1.17.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.1
    # via torch
timm==1.0.12
    # via refiners
tokenizers==0.21.0
    # via transformers
tomli==2.2.1
    # via black
    # via pytest
    # via refiners
torch==2.5.1
    # via bitsandbytes
    # via refiners
    # via segment-anything-hq
    # via segment-anything-py
    # via timm
    # via torchvision
torchvision==0.20.1
    # via piq
    # via refiners
    # via segment-anything-hq
    # via segment-anything-py
    # via timm
tqdm==4.67.1
    # via datasets
    # via huggingface-hub
    # via refiners
    # via transformers
transformers==4.47.1
    # via refiners
triton==3.1.0
    # via torch
types-python-dateutil==2.9.0.20241206
    # via arrow
typing-extensions==4.12.2
    # via bitsandbytes
    # via black
    # via bravado
    # via huggingface-hub
    # via multidict
    # via neptune
    # via pydantic
    # via pydantic-core
    # via swagger-spec-validator
    # via torch
    # via wandb
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.19.1
    # via refiners
watchdog==6.0.0
    # via mkdocs
webcolors==24.11.1
    # via jsonschema
websocket-client==1.8.0
    # via neptune
xxhash==3.5.0
    # via datasets
yarl==1.18.3
    # via aiohttp
zipp==3.21.0
    # via importlib-metadata