How to install for Nvidia 50 series GPUs?

[quasiblob]

You repository readme recommends installing from pypi package.

This didn't work for me at all for some reason, Windows 11 and RTX 50 series GPU.

Can't remember the errors, but I got several and I tried different things until I managed to get this thing installed.

---

If someone else runs into issues with install process, here are the steps, these worked for me at least.
Note - I didn't bother with Flash Attention!

Before install

• Install huggingface CLI, if it is not installed
• Required for gated model download - NOTE - install it for your system Python:
  pip install -U "huggingface_hub[cli]

Install steps

1. Clone the repo
   git clone https://github.com/Stability-AI/stable-audio-tools.git

2. Go to folder
   cd stable-audio-tools

3. Create virtual env (my Python version was 3.10.11, you'll have to have a version installed in your system)
   py -3.10 -m venv venv

4. Activate venv
   Gitbash: source ./venv/scripts/activate or for cmd: call .\venv\scripts\activate

5. Update pip

• I did several install experiments, IIRC I got errors about typing-extensions conflict when installing PyTorch CUDA version, but that didn't happen with newer pip 25.x version:
  python.exe -m pip install --upgrade pip

6. Modify setup.py (NOTE! - this step wasn't required actually, skip it - previous install experiment needed this)
   Open file: notepad setup.py
   Modify line of sentencepiece: sentencepiece==0.2.0

7. Install torch CUDA version (before installing other dependecies)

• Installer does not force exact versions of PyTorch, so I installed RTX 50 suitable version beforehand, as otherwise you get CPU versions:
  pip install torch==2.7.0 torchaudio --index-url https://download.pytorch.org/whl/cu128

8. Then run the setup script:
   pip install .

9. Do a pip check

• It should show 'no broken requirements':
  pip check

Run the app:
python ./run_gradio.py --pretrained-name stabilityai/stable-audio-open-1.0

Note;
I get this warning ConnectionResetError: [WinError 10054] An existing connection was forcibly closed... after every audio clip generation, seems like this happens on Windows and is harmless but it looks ugly in terminal.

[sskalnik]

ConnectionResetError: [WinError 10054] An existing connection was forcibly closed...
I get this on Windows 11 with a 4070 as well. It appears to be a spurious error which you can ignore.

[sskalnik]

I recommend this approach:

pip install --upgrade torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu129

Change the CUDA version string in the URL to whatever you are trying to use. I'm using CUDA 12.9.

[sskalnik]

Note that you will need this on Windows:
dill==0.4.0

I compiled some Flash Attention wheels in case there are still no official wheels for the latest dependencies on Windows 11:
https://github.com/sskalnik/flash_attn_wheels

You won't need these. I just have them because I'm doing some experiments:
bitsandbytes==0.47.0
pipdeptree==2.28.0
torchdata==0.11.0

This is my pipdeptree output:

aiohttp==3.12.15
├── aiohappyeyeballs [required: >=2.5.0, installed: 2.6.1]
├── aiosignal [required: >=1.4.0, installed: 1.4.0]
│   ├── frozenlist [required: >=1.1.0, installed: 1.7.0]
│   └── typing_extensions [required: >=4.2, installed: 4.12.2]
├── attrs [required: >=17.3.0, installed: 25.3.0]
├── frozenlist [required: >=1.1.1, installed: 1.7.0]
├── multidict [required: >=4.5,<7.0, installed: 6.6.4]
├── propcache [required: >=0.2.0, installed: 0.3.2]
└── yarl [required: >=1.17.0,<2.0, installed: 1.20.1]
    ├── idna [required: >=2.0, installed: 3.10]
    ├── multidict [required: >=4.0, installed: 6.6.4]
    └── propcache [required: >=0.2.1, installed: 0.3.2]
altair==5.5.0
├── Jinja2 [required: Any, installed: 3.1.4]
│   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
├── jsonschema [required: >=3.0, installed: 4.25.1]
│   ├── attrs [required: >=22.2.0, installed: 25.3.0]
│   ├── jsonschema-specifications [required: >=2023.03.6, installed: 2025.9.1]
│   │   └── referencing [required: >=0.31.0, installed: 0.36.2]
│   │       ├── attrs [required: >=22.2.0, installed: 25.3.0]
│   │       ├── rpds-py [required: >=0.7.0, installed: 0.27.1]
│   │       └── typing_extensions [required: >=4.4.0, installed: 4.12.2]
│   ├── referencing [required: >=0.28.4, installed: 0.36.2]
│   │   ├── attrs [required: >=22.2.0, installed: 25.3.0]
│   │   ├── rpds-py [required: >=0.7.0, installed: 0.27.1]
│   │   └── typing_extensions [required: >=4.4.0, installed: 4.12.2]
│   └── rpds-py [required: >=0.7.1, installed: 0.27.1]
├── narwhals [required: >=1.14.2, installed: 2.5.0]
├── packaging [required: Any, installed: 25.0]
└── typing_extensions [required: >=4.10.0, installed: 4.12.2]
bitsandbytes==0.47.0
├── torch [required: >=2.2,<3, installed: 2.8.0+cu129]
│   ├── filelock [required: Any, installed: 3.13.1]
│   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   ├── networkx [required: Any, installed: 3.3]
│   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   └── fsspec [required: Any, installed: 2024.6.1]
└── numpy [required: >=1.17, installed: 1.23.5]
Brotli==1.1.0
cmake==4.1.0
dill==0.4.0
flash-attn==2.8.0.post2
├── torch [required: Any, installed: 2.8.0+cu129]
│   ├── filelock [required: Any, installed: 3.13.1]
│   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   ├── networkx [required: Any, installed: 3.3]
│   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   └── fsspec [required: Any, installed: 2024.6.1]
└── einops [required: Any, installed: 0.8.1]
pipdeptree==2.28.0
├── packaging [required: >=24.1, installed: 25.0]
└── pip [required: >=24.2, installed: 25.2]
pytorch_optimizer==3.1.2
├── numpy [required: Any, installed: 1.23.5]
└── torch [required: >=1.10, installed: 2.8.0+cu129]
    ├── filelock [required: Any, installed: 3.13.1]
    ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
    ├── sympy [required: >=1.13.3, installed: 1.13.3]
    │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
    ├── networkx [required: Any, installed: 3.3]
    ├── Jinja2 [required: Any, installed: 3.1.4]
    │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
    └── fsspec [required: Any, installed: 2024.6.1]
stable-audio-tools==0.0.19
├── alias-free-torch [required: ==0.0.6, installed: 0.0.6]
├── auraloss [required: ==0.4.0, installed: 0.4.0]
│   ├── torch [required: Any, installed: 2.8.0+cu129]
│   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   ├── networkx [required: Any, installed: 3.3]
│   │   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   └── fsspec [required: Any, installed: 2024.6.1]
│   └── numpy [required: Any, installed: 1.23.5]
├── descript-audio-codec [required: ==1.0.0, installed: 1.0.0]
│   ├── argbind [required: >=0.3.7, installed: 0.3.9]
│   │   ├── PyYAML [required: Any, installed: 6.0.2]
│   │   └── docstring_parser [required: Any, installed: 0.17.0]
│   ├── descript-audiotools [required: >=0.7.2, installed: 0.7.2]
│   │   ├── argbind [required: Any, installed: 0.3.9]
│   │   │   ├── PyYAML [required: Any, installed: 6.0.2]
│   │   │   └── docstring_parser [required: Any, installed: 0.17.0]
│   │   ├── numpy [required: Any, installed: 1.23.5]
│   │   ├── soundfile [required: Any, installed: 0.13.1]
│   │   │   ├── cffi [required: >=1.0, installed: 2.0.0]
│   │   │   │   └── pycparser [required: Any, installed: 2.23]
│   │   │   └── numpy [required: Any, installed: 1.23.5]
│   │   ├── pyloudnorm [required: Any, installed: 0.1.1]
│   │   │   ├── scipy [required: >=1.0.1, installed: 1.15.3]
│   │   │   │   └── numpy [required: >=1.23.5,<2.5, installed: 1.23.5]
│   │   │   ├── numpy [required: >=1.14.2, installed: 1.23.5]
│   │   │   └── future [required: >=0.16.0, installed: 1.0.0]
│   │   ├── importlib-resources [required: Any, installed: 5.12.0]
│   │   ├── scipy [required: Any, installed: 1.15.3]
│   │   │   └── numpy [required: >=1.23.5,<2.5, installed: 1.23.5]
│   │   ├── torch [required: Any, installed: 2.8.0+cu129]
│   │   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   │   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   │   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   │   ├── networkx [required: Any, installed: 3.3]
│   │   │   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   │   └── fsspec [required: Any, installed: 2024.6.1]
│   │   ├── julius [required: Any, installed: 0.2.7]
│   │   │   └── torch [required: >=1.7.0, installed: 2.8.0+cu129]
│   │   │       ├── filelock [required: Any, installed: 3.13.1]
│   │   │       ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   │       ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │       │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   │       ├── networkx [required: Any, installed: 3.3]
│   │   │       ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │       │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   │       └── fsspec [required: Any, installed: 2024.6.1]
│   │   ├── torchaudio [required: Any, installed: 2.8.0+cu129]
│   │   │   └── torch [required: ==2.8.0+cu129, installed: 2.8.0+cu129]
│   │   │       ├── filelock [required: Any, installed: 3.13.1]
│   │   │       ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   │       ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │       │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   │       ├── networkx [required: Any, installed: 3.3]
│   │   │       ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │       │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   │       └── fsspec [required: Any, installed: 2024.6.1]
│   │   ├── ffmpy [required: Any, installed: 0.6.1]
│   │   ├── ipython [required: Any, installed: 9.5.0]
│   │   │   ├── colorama [required: Any, installed: 0.4.6]
│   │   │   ├── decorator [required: Any, installed: 5.2.1]
│   │   │   ├── ipython_pygments_lexers [required: Any, installed: 1.1.1]
│   │   │   │   └── Pygments [required: Any, installed: 2.19.2]
│   │   │   ├── jedi [required: >=0.16, installed: 0.19.2]
│   │   │   │   └── parso [required: >=0.8.4,<0.9.0, installed: 0.8.5]
│   │   │   ├── matplotlib-inline [required: Any, installed: 0.1.7]
│   │   │   │   └── traitlets [required: Any, installed: 5.14.3]
│   │   │   ├── prompt_toolkit [required: >=3.0.41,<3.1.0, installed: 3.0.52]
│   │   │   │   └── wcwidth [required: Any, installed: 0.2.13]
│   │   │   ├── Pygments [required: >=2.4.0, installed: 2.19.2]
│   │   │   ├── stack-data [required: Any, installed: 0.6.3]
│   │   │   │   ├── executing [required: >=1.2.0, installed: 2.2.1]
│   │   │   │   ├── asttokens [required: >=2.1.0, installed: 3.0.0]
│   │   │   │   └── pure_eval [required: Any, installed: 0.2.3]
│   │   │   ├── traitlets [required: >=5.13.0, installed: 5.14.3]
│   │   │   └── typing_extensions [required: >=4.6, installed: 4.12.2]
│   │   ├── rich [required: Any, installed: 14.1.0]
│   │   │   ├── markdown-it-py [required: >=2.2.0, installed: 4.0.0]
│   │   │   │   └── mdurl [required: ~=0.1, installed: 0.1.2]
│   │   │   └── Pygments [required: >=2.13.0,<3.0.0, installed: 2.19.2]
│   │   ├── matplotlib [required: Any, installed: 3.10.6]
│   │   │   ├── contourpy [required: >=1.0.1, installed: 1.3.2]
│   │   │   │   └── numpy [required: >=1.23, installed: 1.23.5]
│   │   │   ├── cycler [required: >=0.10, installed: 0.12.1]
│   │   │   ├── fonttools [required: >=4.22.0, installed: 4.59.2]
│   │   │   ├── kiwisolver [required: >=1.3.1, installed: 1.4.9]
│   │   │   ├── numpy [required: >=1.23, installed: 1.23.5]
│   │   │   ├── packaging [required: >=20.0, installed: 25.0]
│   │   │   ├── pillow [required: >=8, installed: 10.4.0]
│   │   │   ├── pyparsing [required: >=2.3.1, installed: 3.2.3]
│   │   │   └── python-dateutil [required: >=2.7, installed: 2.9.0.post0]
│   │   │       └── six [required: >=1.5, installed: 1.17.0]
│   │   ├── librosa [required: Any, installed: 0.11.0]
│   │   │   ├── audioread [required: >=2.1.9, installed: 3.0.1]
│   │   │   ├── numba [required: >=0.51.0, installed: 0.60.0]
│   │   │   │   ├── llvmlite [required: >=0.43.0dev0,<0.44, installed: 0.43.0]
│   │   │   │   └── numpy [required: >=1.22,<2.1, installed: 1.23.5]
│   │   │   ├── numpy [required: >=1.22.3, installed: 1.23.5]
│   │   │   ├── scipy [required: >=1.6.0, installed: 1.15.3]
│   │   │   │   └── numpy [required: >=1.23.5,<2.5, installed: 1.23.5]
│   │   │   ├── scikit-learn [required: >=1.1.0, installed: 1.7.2]
│   │   │   │   ├── numpy [required: >=1.22.0, installed: 1.23.5]
│   │   │   │   ├── scipy [required: >=1.8.0, installed: 1.15.3]
│   │   │   │   │   └── numpy [required: >=1.23.5,<2.5, installed: 1.23.5]
│   │   │   │   ├── joblib [required: >=1.2.0, installed: 1.5.2]
│   │   │   │   └── threadpoolctl [required: >=3.1.0, installed: 3.6.0]
│   │   │   ├── joblib [required: >=1.0, installed: 1.5.2]
│   │   │   ├── decorator [required: >=4.3.0, installed: 5.2.1]
│   │   │   ├── soundfile [required: >=0.12.1, installed: 0.13.1]
│   │   │   │   ├── cffi [required: >=1.0, installed: 2.0.0]
│   │   │   │   │   └── pycparser [required: Any, installed: 2.23]
│   │   │   │   └── numpy [required: Any, installed: 1.23.5]
│   │   │   ├── pooch [required: >=1.1, installed: 1.8.2]
│   │   │   │   ├── platformdirs [required: >=2.5.0, installed: 4.4.0]
│   │   │   │   ├── packaging [required: >=20.0, installed: 25.0]
│   │   │   │   └── requests [required: >=2.19.0, installed: 2.32.5]
│   │   │   │       ├── charset-normalizer [required: >=2,<4, installed: 3.4.3]
│   │   │   │       ├── idna [required: >=2.5,<4, installed: 3.10]
│   │   │   │       ├── urllib3 [required: >=1.21.1,<3, installed: 2.5.0]
│   │   │   │       └── certifi [required: >=2017.4.17, installed: 2025.8.3]
│   │   │   ├── soxr [required: >=0.3.2, installed: 1.0.0]
│   │   │   │   └── numpy [required: Any, installed: 1.23.5]
│   │   │   ├── typing_extensions [required: >=4.1.1, installed: 4.12.2]
│   │   │   ├── lazy_loader [required: >=0.1, installed: 0.4]
│   │   │   │   └── packaging [required: Any, installed: 25.0]
│   │   │   └── msgpack [required: >=1.0, installed: 1.1.1]
│   │   ├── pystoi [required: Any, installed: 0.4.1]
│   │   │   ├── numpy [required: Any, installed: 1.23.5]
│   │   │   └── scipy [required: Any, installed: 1.15.3]
│   │   │       └── numpy [required: >=1.23.5,<2.5, installed: 1.23.5]
│   │   ├── torch-stoi [required: Any, installed: 0.2.3]
│   │   │   ├── numpy [required: Any, installed: 1.23.5]
│   │   │   ├── torch [required: Any, installed: 2.8.0+cu129]
│   │   │   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   │   │   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   │   │   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │   │   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   │   │   ├── networkx [required: Any, installed: 3.3]
│   │   │   │   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │   │   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   │   │   └── fsspec [required: Any, installed: 2024.6.1]
│   │   │   ├── pystoi [required: Any, installed: 0.4.1]
│   │   │   │   ├── numpy [required: Any, installed: 1.23.5]
│   │   │   │   └── scipy [required: Any, installed: 1.15.3]
│   │   │   │       └── numpy [required: >=1.23.5,<2.5, installed: 1.23.5]
│   │   │   └── torchaudio [required: Any, installed: 2.8.0+cu129]
│   │   │       └── torch [required: ==2.8.0+cu129, installed: 2.8.0+cu129]
│   │   │           ├── filelock [required: Any, installed: 3.13.1]
│   │   │           ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   │           ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │           │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   │           ├── networkx [required: Any, installed: 3.3]
│   │   │           ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │           │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   │           └── fsspec [required: Any, installed: 2024.6.1]
│   │   ├── flatten-dict [required: Any, installed: 0.4.2]
│   │   │   └── six [required: >=1.12,<2.0, installed: 1.17.0]
│   │   ├── markdown2 [required: Any, installed: 2.5.4]
│   │   ├── randomname [required: Any, installed: 0.2.1]
│   │   │   └── fire [required: Any, installed: 0.7.1]
│   │   │       └── termcolor [required: Any, installed: 3.1.0]
│   │   ├── protobuf [required: >=3.9.2,<3.20, installed: 3.19.6]
│   │   ├── tensorboard [required: Any, installed: 2.20.0]
│   │   │   ├── absl-py [required: >=0.4, installed: 2.3.1]
│   │   │   ├── grpcio [required: >=1.48.2, installed: 1.74.0]
│   │   │   ├── Markdown [required: >=2.6.8, installed: 3.9]
│   │   │   ├── numpy [required: >=1.12.0, installed: 1.23.5]
│   │   │   ├── packaging [required: Any, installed: 25.0]
│   │   │   ├── pillow [required: Any, installed: 10.4.0]
│   │   │   ├── protobuf [required: >=3.19.6,!=4.24.0, installed: 3.19.6]
│   │   │   ├── setuptools [required: >=41.0.0, installed: 80.9.0]
│   │   │   ├── tensorboard-data-server [required: >=0.7.0,<0.8.0, installed: 0.7.2]
│   │   │   └── Werkzeug [required: >=1.0.1, installed: 3.1.3]
│   │   │       └── MarkupSafe [required: >=2.1.1, installed: 2.1.5]
│   │   └── tqdm [required: Any, installed: 4.67.1]
│   │       └── colorama [required: Any, installed: 0.4.6]
│   ├── einops [required: Any, installed: 0.8.1]
│   ├── numpy [required: Any, installed: 1.23.5]
│   ├── torch [required: Any, installed: 2.8.0+cu129]
│   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   ├── networkx [required: Any, installed: 3.3]
│   │   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   └── fsspec [required: Any, installed: 2024.6.1]
│   ├── torchaudio [required: Any, installed: 2.8.0+cu129]
│   │   └── torch [required: ==2.8.0+cu129, installed: 2.8.0+cu129]
│   │       ├── filelock [required: Any, installed: 3.13.1]
│   │       ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │       ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │       │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │       ├── networkx [required: Any, installed: 3.3]
│   │       ├── Jinja2 [required: Any, installed: 3.1.4]
│   │       │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │       └── fsspec [required: Any, installed: 2024.6.1]
│   └── tqdm [required: Any, installed: 4.67.1]
│       └── colorama [required: Any, installed: 0.4.6]
├── einops [required: Any, installed: 0.8.1]
├── einops-exts [required: Any, installed: 0.0.4]
│   └── einops [required: >=0.4, installed: 0.8.1]
├── ema-pytorch [required: ==0.2.3, installed: 0.2.3]
│   └── torch [required: >=1.6, installed: 2.8.0+cu129]
│       ├── filelock [required: Any, installed: 3.13.1]
│       ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│       ├── sympy [required: >=1.13.3, installed: 1.13.3]
│       │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│       ├── networkx [required: Any, installed: 3.3]
│       ├── Jinja2 [required: Any, installed: 3.1.4]
│       │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│       └── fsspec [required: Any, installed: 2024.6.1]
├── encodec [required: ==0.1.1, installed: 0.1.1]
│   ├── numpy [required: Any, installed: 1.23.5]
│   ├── torch [required: Any, installed: 2.8.0+cu129]
│   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   ├── networkx [required: Any, installed: 3.3]
│   │   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   └── fsspec [required: Any, installed: 2024.6.1]
│   ├── torchaudio [required: Any, installed: 2.8.0+cu129]
│   │   └── torch [required: ==2.8.0+cu129, installed: 2.8.0+cu129]
│   │       ├── filelock [required: Any, installed: 3.13.1]
│   │       ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │       ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │       │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │       ├── networkx [required: Any, installed: 3.3]
│   │       ├── Jinja2 [required: Any, installed: 3.1.4]
│   │       │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │       └── fsspec [required: Any, installed: 2024.6.1]
│   └── einops [required: Any, installed: 0.8.1]
├── gradio [required: >=5.20.0, installed: 5.20.0]
│   ├── aiofiles [required: >=22.0,<24.0, installed: 23.2.1]
│   ├── anyio [required: >=3.0,<5.0, installed: 4.10.0]
│   │   ├── idna [required: >=2.8, installed: 3.10]
│   │   ├── sniffio [required: >=1.1, installed: 1.3.1]
│   │   └── typing_extensions [required: >=4.5, installed: 4.12.2]
│   ├── fastapi [required: >=0.115.2,<1.0, installed: 0.116.1]
│   │   ├── starlette [required: >=0.40.0,<0.48.0, installed: 0.47.3]
│   │   │   ├── anyio [required: >=3.6.2,<5, installed: 4.10.0]
│   │   │   │   ├── idna [required: >=2.8, installed: 3.10]
│   │   │   │   ├── sniffio [required: >=1.1, installed: 1.3.1]
│   │   │   │   └── typing_extensions [required: >=4.5, installed: 4.12.2]
│   │   │   └── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   ├── pydantic [required: >=1.7.4,<3.0.0,!=2.1.0,!=2.0.1,!=2.0.0,!=1.8.1,!=1.8, installed: 2.10.6]
│   │   │   ├── annotated-types [required: >=0.6.0, installed: 0.7.0]
│   │   │   ├── pydantic_core [required: ==2.27.2, installed: 2.27.2]
│   │   │   │   └── typing_extensions [required: >=4.6.0,!=4.7.0, installed: 4.12.2]
│   │   │   └── typing_extensions [required: >=4.12.2, installed: 4.12.2]
│   │   └── typing_extensions [required: >=4.8.0, installed: 4.12.2]
│   ├── ffmpy [required: Any, installed: 0.6.1]
│   ├── gradio_client [required: ==1.7.2, installed: 1.7.2]
│   │   ├── fsspec [required: Any, installed: 2024.6.1]
│   │   ├── httpx [required: >=0.24.1, installed: 0.28.1]
│   │   │   ├── anyio [required: Any, installed: 4.10.0]
│   │   │   │   ├── idna [required: >=2.8, installed: 3.10]
│   │   │   │   ├── sniffio [required: >=1.1, installed: 1.3.1]
│   │   │   │   └── typing_extensions [required: >=4.5, installed: 4.12.2]
│   │   │   ├── certifi [required: Any, installed: 2025.8.3]
│   │   │   ├── httpcore [required: ==1.*, installed: 1.0.9]
│   │   │   │   ├── certifi [required: Any, installed: 2025.8.3]
│   │   │   │   └── h11 [required: >=0.16, installed: 0.16.0]
│   │   │   └── idna [required: Any, installed: 3.10]
│   │   ├── huggingface-hub [required: >=0.19.3, installed: 0.34.4]
│   │   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   │   ├── fsspec [required: >=2023.5.0, installed: 2024.6.1]
│   │   │   ├── packaging [required: >=20.9, installed: 25.0]
│   │   │   ├── PyYAML [required: >=5.1, installed: 6.0.2]
│   │   │   ├── requests [required: Any, installed: 2.32.5]
│   │   │   │   ├── charset-normalizer [required: >=2,<4, installed: 3.4.3]
│   │   │   │   ├── idna [required: >=2.5,<4, installed: 3.10]
│   │   │   │   ├── urllib3 [required: >=1.21.1,<3, installed: 2.5.0]
│   │   │   │   └── certifi [required: >=2017.4.17, installed: 2025.8.3]
│   │   │   ├── tqdm [required: >=4.42.1, installed: 4.67.1]
│   │   │   │   └── colorama [required: Any, installed: 0.4.6]
│   │   │   └── typing_extensions [required: >=3.7.4.3, installed: 4.12.2]
│   │   ├── packaging [required: Any, installed: 25.0]
│   │   ├── typing_extensions [required: ~=4.0, installed: 4.12.2]
│   │   └── websockets [required: >=10.0,<16.0, installed: 11.0.3]
│   ├── groovy [required: ~=0.1, installed: 0.1.2]
│   ├── httpx [required: >=0.24.1, installed: 0.28.1]
│   │   ├── anyio [required: Any, installed: 4.10.0]
│   │   │   ├── idna [required: >=2.8, installed: 3.10]
│   │   │   ├── sniffio [required: >=1.1, installed: 1.3.1]
│   │   │   └── typing_extensions [required: >=4.5, installed: 4.12.2]
│   │   ├── certifi [required: Any, installed: 2025.8.3]
│   │   ├── httpcore [required: ==1.*, installed: 1.0.9]
│   │   │   ├── certifi [required: Any, installed: 2025.8.3]
│   │   │   └── h11 [required: >=0.16, installed: 0.16.0]
│   │   └── idna [required: Any, installed: 3.10]
│   ├── huggingface-hub [required: >=0.28.1, installed: 0.34.4]
│   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   ├── fsspec [required: >=2023.5.0, installed: 2024.6.1]
│   │   ├── packaging [required: >=20.9, installed: 25.0]
│   │   ├── PyYAML [required: >=5.1, installed: 6.0.2]
│   │   ├── requests [required: Any, installed: 2.32.5]
│   │   │   ├── charset-normalizer [required: >=2,<4, installed: 3.4.3]
│   │   │   ├── idna [required: >=2.5,<4, installed: 3.10]
│   │   │   ├── urllib3 [required: >=1.21.1,<3, installed: 2.5.0]
│   │   │   └── certifi [required: >=2017.4.17, installed: 2025.8.3]
│   │   ├── tqdm [required: >=4.42.1, installed: 4.67.1]
│   │   │   └── colorama [required: Any, installed: 0.4.6]
│   │   └── typing_extensions [required: >=3.7.4.3, installed: 4.12.2]
│   ├── Jinja2 [required: <4.0, installed: 3.1.4]
│   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   ├── MarkupSafe [required: ~=2.0, installed: 2.1.5]
│   ├── numpy [required: >=1.0,<3.0, installed: 1.23.5]
│   ├── orjson [required: ~=3.0, installed: 3.11.3]
│   ├── packaging [required: Any, installed: 25.0]
│   ├── pandas [required: >=1.0,<3.0, installed: 2.0.2]
│   │   ├── python-dateutil [required: >=2.8.2, installed: 2.9.0.post0]
│   │   │   └── six [required: >=1.5, installed: 1.17.0]
│   │   ├── pytz [required: >=2020.1, installed: 2025.2]
│   │   ├── tzdata [required: >=2022.1, installed: 2025.2]
│   │   ├── numpy [required: >=1.21.0, installed: 1.23.5]
│   │   └── numpy [required: >=1.23.2, installed: 1.23.5]
│   ├── pillow [required: >=8.0,<12.0, installed: 10.4.0]
│   ├── pydantic [required: >=2.0, installed: 2.10.6]
│   │   ├── annotated-types [required: >=0.6.0, installed: 0.7.0]
│   │   ├── pydantic_core [required: ==2.27.2, installed: 2.27.2]
│   │   │   └── typing_extensions [required: >=4.6.0,!=4.7.0, installed: 4.12.2]
│   │   └── typing_extensions [required: >=4.12.2, installed: 4.12.2]
│   ├── pydub [required: Any, installed: 0.25.1]
│   ├── python-multipart [required: >=0.0.18, installed: 0.0.20]
│   ├── PyYAML [required: >=5.0,<7.0, installed: 6.0.2]
│   ├── ruff [required: >=0.9.3, installed: 0.12.12]
│   ├── safehttpx [required: >=0.1.6,<0.2.0, installed: 0.1.6]
│   │   └── httpx [required: Any, installed: 0.28.1]
│   │       ├── anyio [required: Any, installed: 4.10.0]
│   │       │   ├── idna [required: >=2.8, installed: 3.10]
│   │       │   ├── sniffio [required: >=1.1, installed: 1.3.1]
│   │       │   └── typing_extensions [required: >=4.5, installed: 4.12.2]
│   │       ├── certifi [required: Any, installed: 2025.8.3]
│   │       ├── httpcore [required: ==1.*, installed: 1.0.9]
│   │       │   ├── certifi [required: Any, installed: 2025.8.3]
│   │       │   └── h11 [required: >=0.16, installed: 0.16.0]
│   │       └── idna [required: Any, installed: 3.10]
│   ├── semantic-version [required: ~=2.0, installed: 2.10.0]
│   ├── starlette [required: >=0.40.0,<1.0, installed: 0.47.3]
│   │   ├── anyio [required: >=3.6.2,<5, installed: 4.10.0]
│   │   │   ├── idna [required: >=2.8, installed: 3.10]
│   │   │   ├── sniffio [required: >=1.1, installed: 1.3.1]
│   │   │   └── typing_extensions [required: >=4.5, installed: 4.12.2]
│   │   └── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   ├── tomlkit [required: >=0.12.0,<0.14.0, installed: 0.13.3]
│   ├── typer [required: >=0.12,<1.0, installed: 0.17.4]
│   │   ├── click [required: >=8.0.0, installed: 8.2.1]
│   │   │   └── colorama [required: Any, installed: 0.4.6]
│   │   ├── typing_extensions [required: >=3.7.4.3, installed: 4.12.2]
│   │   ├── shellingham [required: >=1.3.0, installed: 1.5.4]
│   │   └── rich [required: >=10.11.0, installed: 14.1.0]
│   │       ├── markdown-it-py [required: >=2.2.0, installed: 4.0.0]
│   │       │   └── mdurl [required: ~=0.1, installed: 0.1.2]
│   │       └── Pygments [required: >=2.13.0,<3.0.0, installed: 2.19.2]
│   ├── typing_extensions [required: ~=4.0, installed: 4.12.2]
│   └── uvicorn [required: >=0.14.0, installed: 0.35.0]
│       ├── click [required: >=7.0, installed: 8.2.1]
│       │   └── colorama [required: Any, installed: 0.4.6]
│       └── h11 [required: >=0.8, installed: 0.16.0]
├── huggingface-hub [required: Any, installed: 0.34.4]
│   ├── filelock [required: Any, installed: 3.13.1]
│   ├── fsspec [required: >=2023.5.0, installed: 2024.6.1]
│   ├── packaging [required: >=20.9, installed: 25.0]
│   ├── PyYAML [required: >=5.1, installed: 6.0.2]
│   ├── requests [required: Any, installed: 2.32.5]
│   │   ├── charset-normalizer [required: >=2,<4, installed: 3.4.3]
│   │   ├── idna [required: >=2.5,<4, installed: 3.10]
│   │   ├── urllib3 [required: >=1.21.1,<3, installed: 2.5.0]
│   │   └── certifi [required: >=2017.4.17, installed: 2025.8.3]
│   ├── tqdm [required: >=4.42.1, installed: 4.67.1]
│   │   └── colorama [required: Any, installed: 0.4.6]
│   └── typing_extensions [required: >=3.7.4.3, installed: 4.12.2]
├── importlib-resources [required: ==5.12.0, installed: 5.12.0]
├── k-diffusion [required: ==0.1.1, installed: 0.1.1]
│   ├── accelerate [required: Any, installed: 1.10.1]
│   │   ├── numpy [required: >=1.17,<3.0.0, installed: 1.23.5]
│   │   ├── packaging [required: >=20.0, installed: 25.0]
│   │   ├── psutil [required: Any, installed: 7.0.0]
│   │   ├── PyYAML [required: Any, installed: 6.0.2]
│   │   ├── torch [required: >=2.0.0, installed: 2.8.0+cu129]
│   │   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   │   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   │   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   │   ├── networkx [required: Any, installed: 3.3]
│   │   │   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   │   └── fsspec [required: Any, installed: 2024.6.1]
│   │   ├── huggingface-hub [required: >=0.21.0, installed: 0.34.4]
│   │   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   │   ├── fsspec [required: >=2023.5.0, installed: 2024.6.1]
│   │   │   ├── packaging [required: >=20.9, installed: 25.0]
│   │   │   ├── PyYAML [required: >=5.1, installed: 6.0.2]
│   │   │   ├── requests [required: Any, installed: 2.32.5]
│   │   │   │   ├── charset-normalizer [required: >=2,<4, installed: 3.4.3]
│   │   │   │   ├── idna [required: >=2.5,<4, installed: 3.10]
│   │   │   │   ├── urllib3 [required: >=1.21.1,<3, installed: 2.5.0]
│   │   │   │   └── certifi [required: >=2017.4.17, installed: 2025.8.3]
│   │   │   ├── tqdm [required: >=4.42.1, installed: 4.67.1]
│   │   │   │   └── colorama [required: Any, installed: 0.4.6]
│   │   │   └── typing_extensions [required: >=3.7.4.3, installed: 4.12.2]
│   │   └── safetensors [required: >=0.4.3, installed: 0.6.2]
│   ├── clean-fid [required: Any, installed: 0.1.35]
│   │   ├── torch [required: Any, installed: 2.8.0+cu129]
│   │   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   │   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   │   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   │   ├── networkx [required: Any, installed: 3.3]
│   │   │   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   │   └── fsspec [required: Any, installed: 2024.6.1]
│   │   ├── torchvision [required: Any, installed: 0.23.0+cu129]
│   │   │   ├── numpy [required: Any, installed: 1.23.5]
│   │   │   ├── torch [required: ==2.8.0+cu129, installed: 2.8.0+cu129]
│   │   │   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   │   │   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   │   │   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │   │   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   │   │   ├── networkx [required: Any, installed: 3.3]
│   │   │   │   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │   │   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   │   │   └── fsspec [required: Any, installed: 2024.6.1]
│   │   │   └── pillow [required: >=5.3.0,!=8.3.*, installed: 10.4.0]
│   │   ├── numpy [required: >=1.14.3, installed: 1.23.5]
│   │   ├── scipy [required: >=1.0.1, installed: 1.15.3]
│   │   │   └── numpy [required: >=1.23.5,<2.5, installed: 1.23.5]
│   │   ├── tqdm [required: >=4.28.1, installed: 4.67.1]
│   │   │   └── colorama [required: Any, installed: 0.4.6]
│   │   ├── pillow [required: >=8.1, installed: 10.4.0]
│   │   └── requests [required: Any, installed: 2.32.5]
│   │       ├── charset-normalizer [required: >=2,<4, installed: 3.4.3]
│   │       ├── idna [required: >=2.5,<4, installed: 3.10]
│   │       ├── urllib3 [required: >=1.21.1,<3, installed: 2.5.0]
│   │       └── certifi [required: >=2017.4.17, installed: 2025.8.3]
│   ├── clip-anytorch [required: Any, installed: 2.6.0]
│   │   ├── ftfy [required: Any, installed: 6.3.1]
│   │   │   └── wcwidth [required: Any, installed: 0.2.13]
│   │   ├── regex [required: Any, installed: 2025.9.1]
│   │   ├── tqdm [required: Any, installed: 4.67.1]
│   │   │   └── colorama [required: Any, installed: 0.4.6]
│   │   ├── torch [required: Any, installed: 2.8.0+cu129]
│   │   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   │   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   │   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   │   ├── networkx [required: Any, installed: 3.3]
│   │   │   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   │   └── fsspec [required: Any, installed: 2024.6.1]
│   │   └── torchvision [required: Any, installed: 0.23.0+cu129]
│   │       ├── numpy [required: Any, installed: 1.23.5]
│   │       ├── torch [required: ==2.8.0+cu129, installed: 2.8.0+cu129]
│   │       │   ├── filelock [required: Any, installed: 3.13.1]
│   │       │   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │       │   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │       │   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │       │   ├── networkx [required: Any, installed: 3.3]
│   │       │   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │       │   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │       │   └── fsspec [required: Any, installed: 2024.6.1]
│   │       └── pillow [required: >=5.3.0,!=8.3.*, installed: 10.4.0]
│   ├── dctorch [required: Any, installed: 0.1.2]
│   │   ├── numpy [required: >=1.22.3,<2.0.0, installed: 1.23.5]
│   │   ├── scipy [required: >=1.8.0,<2.0.0, installed: 1.15.3]
│   │   │   └── numpy [required: >=1.23.5,<2.5, installed: 1.23.5]
│   │   └── torch [required: >=1.11.0,<3.0, installed: 2.8.0+cu129]
│   │       ├── filelock [required: Any, installed: 3.13.1]
│   │       ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │       ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │       │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │       ├── networkx [required: Any, installed: 3.3]
│   │       ├── Jinja2 [required: Any, installed: 3.1.4]
│   │       │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │       └── fsspec [required: Any, installed: 2024.6.1]
│   ├── einops [required: Any, installed: 0.8.1]
│   ├── jsonmerge [required: Any, installed: 1.9.2]
│   │   └── jsonschema [required: >2.4.0, installed: 4.25.1]
│   │       ├── attrs [required: >=22.2.0, installed: 25.3.0]
│   │       ├── jsonschema-specifications [required: >=2023.03.6, installed: 2025.9.1]
│   │       │   └── referencing [required: >=0.31.0, installed: 0.36.2]
│   │       │       ├── attrs [required: >=22.2.0, installed: 25.3.0]
│   │       │       ├── rpds-py [required: >=0.7.0, installed: 0.27.1]
│   │       │       └── typing_extensions [required: >=4.4.0, installed: 4.12.2]
│   │       ├── referencing [required: >=0.28.4, installed: 0.36.2]
│   │       │   ├── attrs [required: >=22.2.0, installed: 25.3.0]
│   │       │   ├── rpds-py [required: >=0.7.0, installed: 0.27.1]
│   │       │   └── typing_extensions [required: >=4.4.0, installed: 4.12.2]
│   │       └── rpds-py [required: >=0.7.1, installed: 0.27.1]
│   ├── kornia [required: Any, installed: 0.8.1]
│   │   ├── kornia_rs [required: >=0.1.9, installed: 0.1.9]
│   │   ├── packaging [required: Any, installed: 25.0]
│   │   └── torch [required: >=1.9.1, installed: 2.8.0+cu129]
│   │       ├── filelock [required: Any, installed: 3.13.1]
│   │       ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │       ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │       │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │       ├── networkx [required: Any, installed: 3.3]
│   │       ├── Jinja2 [required: Any, installed: 3.1.4]
│   │       │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │       └── fsspec [required: Any, installed: 2024.6.1]
│   ├── pillow [required: Any, installed: 10.4.0]
│   ├── safetensors [required: Any, installed: 0.6.2]
│   ├── scikit-image [required: Any, installed: 0.24.0]
│   │   ├── numpy [required: >=1.23, installed: 1.23.5]
│   │   ├── scipy [required: >=1.9, installed: 1.15.3]
│   │   │   └── numpy [required: >=1.23.5,<2.5, installed: 1.23.5]
│   │   ├── networkx [required: >=2.8, installed: 3.3]
│   │   ├── pillow [required: >=9.1, installed: 10.4.0]
│   │   ├── imageio [required: >=2.33, installed: 2.37.0]
│   │   │   ├── numpy [required: Any, installed: 1.23.5]
│   │   │   └── pillow [required: >=8.3.2, installed: 10.4.0]
│   │   ├── tifffile [required: >=2022.8.12, installed: 2025.9.9]
│   │   │   └── numpy [required: Any, installed: 1.23.5]
│   │   ├── packaging [required: >=21, installed: 25.0]
│   │   └── lazy_loader [required: >=0.4, installed: 0.4]
│   │       └── packaging [required: Any, installed: 25.0]
│   ├── scipy [required: Any, installed: 1.15.3]
│   │   └── numpy [required: >=1.23.5,<2.5, installed: 1.23.5]
│   ├── torch [required: >=2.0, installed: 2.8.0+cu129]
│   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   ├── networkx [required: Any, installed: 3.3]
│   │   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   └── fsspec [required: Any, installed: 2024.6.1]
│   ├── torchdiffeq [required: Any, installed: 0.2.5]
│   │   ├── torch [required: >=1.5.0, installed: 2.8.0+cu129]
│   │   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   │   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   │   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   │   ├── networkx [required: Any, installed: 3.3]
│   │   │   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   │   └── fsspec [required: Any, installed: 2024.6.1]
│   │   └── scipy [required: >=1.4.0, installed: 1.15.3]
│   │       └── numpy [required: >=1.23.5,<2.5, installed: 1.23.5]
│   ├── torchsde [required: Any, installed: 0.2.6]
│   │   ├── numpy [required: >=1.19, installed: 1.23.5]
│   │   ├── scipy [required: >=1.5, installed: 1.15.3]
│   │   │   └── numpy [required: >=1.23.5,<2.5, installed: 1.23.5]
│   │   ├── torch [required: >=1.6.0, installed: 2.8.0+cu129]
│   │   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   │   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   │   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   │   ├── networkx [required: Any, installed: 3.3]
│   │   │   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   │   └── fsspec [required: Any, installed: 2024.6.1]
│   │   └── trampoline [required: >=0.1.2, installed: 0.1.2]
│   ├── torchvision [required: Any, installed: 0.23.0+cu129]
│   │   ├── numpy [required: Any, installed: 1.23.5]
│   │   ├── torch [required: ==2.8.0+cu129, installed: 2.8.0+cu129]
│   │   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   │   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   │   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   │   ├── networkx [required: Any, installed: 3.3]
│   │   │   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   │   └── fsspec [required: Any, installed: 2024.6.1]
│   │   └── pillow [required: >=5.3.0,!=8.3.*, installed: 10.4.0]
│   ├── tqdm [required: Any, installed: 4.67.1]
│   │   └── colorama [required: Any, installed: 0.4.6]
│   └── wandb [required: Any, installed: 0.15.4]
│       ├── click [required: >=7.0,!=8.0.0, installed: 8.2.1]
│       │   └── colorama [required: Any, installed: 0.4.6]
│       ├── GitPython [required: >=1.0.0,!=3.1.29, installed: 3.1.45]
│       │   └── gitdb [required: >=4.0.1,<5, installed: 4.0.12]
│       │       └── smmap [required: >=3.0.1,<6, installed: 5.0.2]
│       ├── requests [required: >=2.0.0,<3, installed: 2.32.5]
│       │   ├── charset-normalizer [required: >=2,<4, installed: 3.4.3]
│       │   ├── idna [required: >=2.5,<4, installed: 3.10]
│       │   ├── urllib3 [required: >=1.21.1,<3, installed: 2.5.0]
│       │   └── certifi [required: >=2017.4.17, installed: 2025.8.3]
│       ├── psutil [required: >=5.0.0, installed: 7.0.0]
│       ├── sentry-sdk [required: >=1.0.0, installed: 2.37.1]
│       │   ├── urllib3 [required: >=1.26.11, installed: 2.5.0]
│       │   └── certifi [required: Any, installed: 2025.8.3]
│       ├── docker-pycreds [required: >=0.4.0, installed: 0.4.0]
│       │   └── six [required: >=1.4.0, installed: 1.17.0]
│       ├── PyYAML [required: Any, installed: 6.0.2]
│       ├── pathtools [required: Any, installed: 0.1.2]
│       ├── setproctitle [required: Any, installed: 1.3.7]
│       ├── setuptools [required: Any, installed: 80.9.0]
│       ├── appdirs [required: >=1.4.3, installed: 1.4.4]
│       └── protobuf [required: >=3.19.0,<5,!=4.21.0, installed: 3.19.6]
├── laion-clap [required: ==1.1.4, installed: 1.1.4]
│   ├── numpy [required: ==1.23.5, installed: 1.23.5]
│   ├── soundfile [required: Any, installed: 0.13.1]
│   │   ├── cffi [required: >=1.0, installed: 2.0.0]
│   │   │   └── pycparser [required: Any, installed: 2.23]
│   │   └── numpy [required: Any, installed: 1.23.5]
│   ├── librosa [required: Any, installed: 0.11.0]
│   │   ├── audioread [required: >=2.1.9, installed: 3.0.1]
│   │   ├── numba [required: >=0.51.0, installed: 0.60.0]
│   │   │   ├── llvmlite [required: >=0.43.0dev0,<0.44, installed: 0.43.0]
│   │   │   └── numpy [required: >=1.22,<2.1, installed: 1.23.5]
│   │   ├── numpy [required: >=1.22.3, installed: 1.23.5]
│   │   ├── scipy [required: >=1.6.0, installed: 1.15.3]
│   │   │   └── numpy [required: >=1.23.5,<2.5, installed: 1.23.5]
│   │   ├── scikit-learn [required: >=1.1.0, installed: 1.7.2]
│   │   │   ├── numpy [required: >=1.22.0, installed: 1.23.5]
│   │   │   ├── scipy [required: >=1.8.0, installed: 1.15.3]
│   │   │   │   └── numpy [required: >=1.23.5,<2.5, installed: 1.23.5]
│   │   │   ├── joblib [required: >=1.2.0, installed: 1.5.2]
│   │   │   └── threadpoolctl [required: >=3.1.0, installed: 3.6.0]
│   │   ├── joblib [required: >=1.0, installed: 1.5.2]
│   │   ├── decorator [required: >=4.3.0, installed: 5.2.1]
│   │   ├── soundfile [required: >=0.12.1, installed: 0.13.1]
│   │   │   ├── cffi [required: >=1.0, installed: 2.0.0]
│   │   │   │   └── pycparser [required: Any, installed: 2.23]
│   │   │   └── numpy [required: Any, installed: 1.23.5]
│   │   ├── pooch [required: >=1.1, installed: 1.8.2]
│   │   │   ├── platformdirs [required: >=2.5.0, installed: 4.4.0]
│   │   │   ├── packaging [required: >=20.0, installed: 25.0]
│   │   │   └── requests [required: >=2.19.0, installed: 2.32.5]
│   │   │       ├── charset-normalizer [required: >=2,<4, installed: 3.4.3]
│   │   │       ├── idna [required: >=2.5,<4, installed: 3.10]
│   │   │       ├── urllib3 [required: >=1.21.1,<3, installed: 2.5.0]
│   │   │       └── certifi [required: >=2017.4.17, installed: 2025.8.3]
│   │   ├── soxr [required: >=0.3.2, installed: 1.0.0]
│   │   │   └── numpy [required: Any, installed: 1.23.5]
│   │   ├── typing_extensions [required: >=4.1.1, installed: 4.12.2]
│   │   ├── lazy_loader [required: >=0.1, installed: 0.4]
│   │   │   └── packaging [required: Any, installed: 25.0]
│   │   └── msgpack [required: >=1.0, installed: 1.1.1]
│   ├── torchlibrosa [required: Any, installed: 0.1.0]
│   │   ├── numpy [required: Any, installed: 1.23.5]
│   │   └── librosa [required: >=0.8.0, installed: 0.11.0]
│   │       ├── audioread [required: >=2.1.9, installed: 3.0.1]
│   │       ├── numba [required: >=0.51.0, installed: 0.60.0]
│   │       │   ├── llvmlite [required: >=0.43.0dev0,<0.44, installed: 0.43.0]
│   │       │   └── numpy [required: >=1.22,<2.1, installed: 1.23.5]
│   │       ├── numpy [required: >=1.22.3, installed: 1.23.5]
│   │       ├── scipy [required: >=1.6.0, installed: 1.15.3]
│   │       │   └── numpy [required: >=1.23.5,<2.5, installed: 1.23.5]
│   │       ├── scikit-learn [required: >=1.1.0, installed: 1.7.2]
│   │       │   ├── numpy [required: >=1.22.0, installed: 1.23.5]
│   │       │   ├── scipy [required: >=1.8.0, installed: 1.15.3]
│   │       │   │   └── numpy [required: >=1.23.5,<2.5, installed: 1.23.5]
│   │       │   ├── joblib [required: >=1.2.0, installed: 1.5.2]
│   │       │   └── threadpoolctl [required: >=3.1.0, installed: 3.6.0]
│   │       ├── joblib [required: >=1.0, installed: 1.5.2]
│   │       ├── decorator [required: >=4.3.0, installed: 5.2.1]
│   │       ├── soundfile [required: >=0.12.1, installed: 0.13.1]
│   │       │   ├── cffi [required: >=1.0, installed: 2.0.0]
│   │       │   │   └── pycparser [required: Any, installed: 2.23]
│   │       │   └── numpy [required: Any, installed: 1.23.5]
│   │       ├── pooch [required: >=1.1, installed: 1.8.2]
│   │       │   ├── platformdirs [required: >=2.5.0, installed: 4.4.0]
│   │       │   ├── packaging [required: >=20.0, installed: 25.0]
│   │       │   └── requests [required: >=2.19.0, installed: 2.32.5]
│   │       │       ├── charset-normalizer [required: >=2,<4, installed: 3.4.3]
│   │       │       ├── idna [required: >=2.5,<4, installed: 3.10]
│   │       │       ├── urllib3 [required: >=1.21.1,<3, installed: 2.5.0]
│   │       │       └── certifi [required: >=2017.4.17, installed: 2025.8.3]
│   │       ├── soxr [required: >=0.3.2, installed: 1.0.0]
│   │       │   └── numpy [required: Any, installed: 1.23.5]
│   │       ├── typing_extensions [required: >=4.1.1, installed: 4.12.2]
│   │       ├── lazy_loader [required: >=0.1, installed: 0.4]
│   │       │   └── packaging [required: Any, installed: 25.0]
│   │       └── msgpack [required: >=1.0, installed: 1.1.1]
│   ├── ftfy [required: Any, installed: 6.3.1]
│   │   └── wcwidth [required: Any, installed: 0.2.13]
│   ├── braceexpand [required: Any, installed: 0.1.7]
│   ├── webdataset [required: Any, installed: 0.2.100]
│   │   ├── braceexpand [required: Any, installed: 0.1.7]
│   │   ├── numpy [required: Any, installed: 1.23.5]
│   │   └── PyYAML [required: Any, installed: 6.0.2]
│   ├── wget [required: Any, installed: 3.2]
│   ├── wandb [required: Any, installed: 0.15.4]
│   │   ├── click [required: >=7.0,!=8.0.0, installed: 8.2.1]
│   │   │   └── colorama [required: Any, installed: 0.4.6]
│   │   ├── GitPython [required: >=1.0.0,!=3.1.29, installed: 3.1.45]
│   │   │   └── gitdb [required: >=4.0.1,<5, installed: 4.0.12]
│   │   │       └── smmap [required: >=3.0.1,<6, installed: 5.0.2]
│   │   ├── requests [required: >=2.0.0,<3, installed: 2.32.5]
│   │   │   ├── charset-normalizer [required: >=2,<4, installed: 3.4.3]
│   │   │   ├── idna [required: >=2.5,<4, installed: 3.10]
│   │   │   ├── urllib3 [required: >=1.21.1,<3, installed: 2.5.0]
│   │   │   └── certifi [required: >=2017.4.17, installed: 2025.8.3]
│   │   ├── psutil [required: >=5.0.0, installed: 7.0.0]
│   │   ├── sentry-sdk [required: >=1.0.0, installed: 2.37.1]
│   │   │   ├── urllib3 [required: >=1.26.11, installed: 2.5.0]
│   │   │   └── certifi [required: Any, installed: 2025.8.3]
│   │   ├── docker-pycreds [required: >=0.4.0, installed: 0.4.0]
│   │   │   └── six [required: >=1.4.0, installed: 1.17.0]
│   │   ├── PyYAML [required: Any, installed: 6.0.2]
│   │   ├── pathtools [required: Any, installed: 0.1.2]
│   │   ├── setproctitle [required: Any, installed: 1.3.7]
│   │   ├── setuptools [required: Any, installed: 80.9.0]
│   │   ├── appdirs [required: >=1.4.3, installed: 1.4.4]
│   │   └── protobuf [required: >=3.19.0,<5,!=4.21.0, installed: 3.19.6]
│   ├── llvmlite [required: Any, installed: 0.43.0]
│   ├── scipy [required: Any, installed: 1.15.3]
│   │   └── numpy [required: >=1.23.5,<2.5, installed: 1.23.5]
│   ├── scikit-learn [required: Any, installed: 1.7.2]
│   │   ├── numpy [required: >=1.22.0, installed: 1.23.5]
│   │   ├── scipy [required: >=1.8.0, installed: 1.15.3]
│   │   │   └── numpy [required: >=1.23.5,<2.5, installed: 1.23.5]
│   │   ├── joblib [required: >=1.2.0, installed: 1.5.2]
│   │   └── threadpoolctl [required: >=3.1.0, installed: 3.6.0]
│   ├── pandas [required: Any, installed: 2.0.2]
│   │   ├── python-dateutil [required: >=2.8.2, installed: 2.9.0.post0]
│   │   │   └── six [required: >=1.5, installed: 1.17.0]
│   │   ├── pytz [required: >=2020.1, installed: 2025.2]
│   │   ├── tzdata [required: >=2022.1, installed: 2025.2]
│   │   ├── numpy [required: >=1.21.0, installed: 1.23.5]
│   │   └── numpy [required: >=1.23.2, installed: 1.23.5]
│   ├── h5py [required: Any, installed: 3.14.0]
│   │   └── numpy [required: >=1.19.3, installed: 1.23.5]
│   ├── tqdm [required: Any, installed: 4.67.1]
│   │   └── colorama [required: Any, installed: 0.4.6]
│   ├── regex [required: Any, installed: 2025.9.1]
│   ├── transformers [required: Any, installed: 4.56.1]
│   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   ├── huggingface-hub [required: >=0.34.0,<1.0, installed: 0.34.4]
│   │   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   │   ├── fsspec [required: >=2023.5.0, installed: 2024.6.1]
│   │   │   ├── packaging [required: >=20.9, installed: 25.0]
│   │   │   ├── PyYAML [required: >=5.1, installed: 6.0.2]
│   │   │   ├── requests [required: Any, installed: 2.32.5]
│   │   │   │   ├── charset-normalizer [required: >=2,<4, installed: 3.4.3]
│   │   │   │   ├── idna [required: >=2.5,<4, installed: 3.10]
│   │   │   │   ├── urllib3 [required: >=1.21.1,<3, installed: 2.5.0]
│   │   │   │   └── certifi [required: >=2017.4.17, installed: 2025.8.3]
│   │   │   ├── tqdm [required: >=4.42.1, installed: 4.67.1]
│   │   │   │   └── colorama [required: Any, installed: 0.4.6]
│   │   │   └── typing_extensions [required: >=3.7.4.3, installed: 4.12.2]
│   │   ├── numpy [required: >=1.17, installed: 1.23.5]
│   │   ├── packaging [required: >=20.0, installed: 25.0]
│   │   ├── PyYAML [required: >=5.1, installed: 6.0.2]
│   │   ├── regex [required: !=2019.12.17, installed: 2025.9.1]
│   │   ├── requests [required: Any, installed: 2.32.5]
│   │   │   ├── charset-normalizer [required: >=2,<4, installed: 3.4.3]
│   │   │   ├── idna [required: >=2.5,<4, installed: 3.10]
│   │   │   ├── urllib3 [required: >=1.21.1,<3, installed: 2.5.0]
│   │   │   └── certifi [required: >=2017.4.17, installed: 2025.8.3]
│   │   ├── tokenizers [required: >=0.22.0,<=0.23.0, installed: 0.22.0]
│   │   │   └── huggingface-hub [required: >=0.16.4,<1.0, installed: 0.34.4]
│   │   │       ├── filelock [required: Any, installed: 3.13.1]
│   │   │       ├── fsspec [required: >=2023.5.0, installed: 2024.6.1]
│   │   │       ├── packaging [required: >=20.9, installed: 25.0]
│   │   │       ├── PyYAML [required: >=5.1, installed: 6.0.2]
│   │   │       ├── requests [required: Any, installed: 2.32.5]
│   │   │       │   ├── charset-normalizer [required: >=2,<4, installed: 3.4.3]
│   │   │       │   ├── idna [required: >=2.5,<4, installed: 3.10]
│   │   │       │   ├── urllib3 [required: >=1.21.1,<3, installed: 2.5.0]
│   │   │       │   └── certifi [required: >=2017.4.17, installed: 2025.8.3]
│   │   │       ├── tqdm [required: >=4.42.1, installed: 4.67.1]
│   │   │       │   └── colorama [required: Any, installed: 0.4.6]
│   │   │       └── typing_extensions [required: >=3.7.4.3, installed: 4.12.2]
│   │   ├── safetensors [required: >=0.4.3, installed: 0.6.2]
│   │   └── tqdm [required: >=4.27, installed: 4.67.1]
│   │       └── colorama [required: Any, installed: 0.4.6]
│   └── progressbar [required: Any, installed: 2.5]
├── local-attention [required: ==1.8.6, installed: 1.8.6]
│   ├── einops [required: >=0.6.0, installed: 0.8.1]
│   └── torch [required: Any, installed: 2.8.0+cu129]
│       ├── filelock [required: Any, installed: 3.13.1]
│       ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│       ├── sympy [required: >=1.13.3, installed: 1.13.3]
│       │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│       ├── networkx [required: Any, installed: 3.3]
│       ├── Jinja2 [required: Any, installed: 3.1.4]
│       │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│       └── fsspec [required: Any, installed: 2024.6.1]
├── pandas [required: ==2.0.2, installed: 2.0.2]
│   ├── python-dateutil [required: >=2.8.2, installed: 2.9.0.post0]
│   │   └── six [required: >=1.5, installed: 1.17.0]
│   ├── pytz [required: >=2020.1, installed: 2025.2]
│   ├── tzdata [required: >=2022.1, installed: 2025.2]
│   ├── numpy [required: >=1.21.0, installed: 1.23.5]
│   └── numpy [required: >=1.23.2, installed: 1.23.5]
├── prefigure [required: ==0.0.9, installed: 0.0.9]
│   ├── argparse [required: Any, installed: 1.4.0]
│   ├── configparser [required: Any, installed: 7.2.0]
│   ├── wandb [required: Any, installed: 0.15.4]
│   │   ├── click [required: >=7.0,!=8.0.0, installed: 8.2.1]
│   │   │   └── colorama [required: Any, installed: 0.4.6]
│   │   ├── GitPython [required: >=1.0.0,!=3.1.29, installed: 3.1.45]
│   │   │   └── gitdb [required: >=4.0.1,<5, installed: 4.0.12]
│   │   │       └── smmap [required: >=3.0.1,<6, installed: 5.0.2]
│   │   ├── requests [required: >=2.0.0,<3, installed: 2.32.5]
│   │   │   ├── charset-normalizer [required: >=2,<4, installed: 3.4.3]
│   │   │   ├── idna [required: >=2.5,<4, installed: 3.10]
│   │   │   ├── urllib3 [required: >=1.21.1,<3, installed: 2.5.0]
│   │   │   └── certifi [required: >=2017.4.17, installed: 2025.8.3]
│   │   ├── psutil [required: >=5.0.0, installed: 7.0.0]
│   │   ├── sentry-sdk [required: >=1.0.0, installed: 2.37.1]
│   │   │   ├── urllib3 [required: >=1.26.11, installed: 2.5.0]
│   │   │   └── certifi [required: Any, installed: 2025.8.3]
│   │   ├── docker-pycreds [required: >=0.4.0, installed: 0.4.0]
│   │   │   └── six [required: >=1.4.0, installed: 1.17.0]
│   │   ├── PyYAML [required: Any, installed: 6.0.2]
│   │   ├── pathtools [required: Any, installed: 0.1.2]
│   │   ├── setproctitle [required: Any, installed: 1.3.7]
│   │   ├── setuptools [required: Any, installed: 80.9.0]
│   │   ├── appdirs [required: >=1.4.3, installed: 1.4.4]
│   │   └── protobuf [required: >=3.19.0,<5,!=4.21.0, installed: 3.19.6]
│   ├── pytorch-lightning [required: Any, installed: 2.4.0]
│   │   ├── torch [required: >=2.1.0, installed: 2.8.0+cu129]
│   │   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   │   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   │   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   │   ├── networkx [required: Any, installed: 3.3]
│   │   │   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   │   └── fsspec [required: Any, installed: 2024.6.1]
│   │   ├── tqdm [required: >=4.57.0, installed: 4.67.1]
│   │   │   └── colorama [required: Any, installed: 0.4.6]
│   │   ├── PyYAML [required: >=5.4, installed: 6.0.2]
│   │   ├── fsspec [required: >=2022.5.0, installed: 2024.6.1]
│   │   ├── torchmetrics [required: >=0.7.0, installed: 0.11.4]
│   │   │   ├── numpy [required: >=1.17.2, installed: 1.23.5]
│   │   │   ├── torch [required: >=1.8.1, installed: 2.8.0+cu129]
│   │   │   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   │   │   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   │   │   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │   │   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   │   │   ├── networkx [required: Any, installed: 3.3]
│   │   │   │   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │   │   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   │   │   └── fsspec [required: Any, installed: 2024.6.1]
│   │   │   └── packaging [required: Any, installed: 25.0]
│   │   ├── packaging [required: >=20.0, installed: 25.0]
│   │   ├── typing_extensions [required: >=4.4.0, installed: 4.12.2]
│   │   └── lightning-utilities [required: >=0.10.0, installed: 0.15.2]
│   │       ├── packaging [required: >=17.1, installed: 25.0]
│   │       ├── setuptools [required: Any, installed: 80.9.0]
│   │       └── typing_extensions [required: Any, installed: 4.12.2]
│   └── gin-config [required: Any, installed: 0.5.0]
├── pytorch-lightning [required: ==2.4.0, installed: 2.4.0]
│   ├── torch [required: >=2.1.0, installed: 2.8.0+cu129]
│   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   ├── networkx [required: Any, installed: 3.3]
│   │   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   └── fsspec [required: Any, installed: 2024.6.1]
│   ├── tqdm [required: >=4.57.0, installed: 4.67.1]
│   │   └── colorama [required: Any, installed: 0.4.6]
│   ├── PyYAML [required: >=5.4, installed: 6.0.2]
│   ├── fsspec [required: >=2022.5.0, installed: 2024.6.1]
│   ├── torchmetrics [required: >=0.7.0, installed: 0.11.4]
│   │   ├── numpy [required: >=1.17.2, installed: 1.23.5]
│   │   ├── torch [required: >=1.8.1, installed: 2.8.0+cu129]
│   │   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   │   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   │   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   │   ├── networkx [required: Any, installed: 3.3]
│   │   │   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   │   └── fsspec [required: Any, installed: 2024.6.1]
│   │   └── packaging [required: Any, installed: 25.0]
│   ├── packaging [required: >=20.0, installed: 25.0]
│   ├── typing_extensions [required: >=4.4.0, installed: 4.12.2]
│   └── lightning-utilities [required: >=0.10.0, installed: 0.15.2]
│       ├── packaging [required: >=17.1, installed: 25.0]
│       ├── setuptools [required: Any, installed: 80.9.0]
│       └── typing_extensions [required: Any, installed: 4.12.2]
├── PyWavelets [required: ==1.4.1, installed: 1.4.1]
│   └── numpy [required: >=1.17.3, installed: 1.23.5]
├── safetensors [required: Any, installed: 0.6.2]
├── sentencepiece [required: ==0.1.99, installed: 0.1.99]
├── torch [required: >=2.5.1, installed: 2.8.0+cu129]
│   ├── filelock [required: Any, installed: 3.13.1]
│   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   ├── networkx [required: Any, installed: 3.3]
│   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   └── fsspec [required: Any, installed: 2024.6.1]
├── torchaudio [required: >=2.5.1, installed: 2.8.0+cu129]
│   └── torch [required: ==2.8.0+cu129, installed: 2.8.0+cu129]
│       ├── filelock [required: Any, installed: 3.13.1]
│       ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│       ├── sympy [required: >=1.13.3, installed: 1.13.3]
│       │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│       ├── networkx [required: Any, installed: 3.3]
│       ├── Jinja2 [required: Any, installed: 3.1.4]
│       │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│       └── fsspec [required: Any, installed: 2024.6.1]
├── torchmetrics [required: ==0.11.4, installed: 0.11.4]
│   ├── numpy [required: >=1.17.2, installed: 1.23.5]
│   ├── torch [required: >=1.8.1, installed: 2.8.0+cu129]
│   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   ├── networkx [required: Any, installed: 3.3]
│   │   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   └── fsspec [required: Any, installed: 2024.6.1]
│   └── packaging [required: Any, installed: 25.0]
├── tqdm [required: Any, installed: 4.67.1]
│   └── colorama [required: Any, installed: 0.4.6]
├── transformers [required: Any, installed: 4.56.1]
│   ├── filelock [required: Any, installed: 3.13.1]
│   ├── huggingface-hub [required: >=0.34.0,<1.0, installed: 0.34.4]
│   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   ├── fsspec [required: >=2023.5.0, installed: 2024.6.1]
│   │   ├── packaging [required: >=20.9, installed: 25.0]
│   │   ├── PyYAML [required: >=5.1, installed: 6.0.2]
│   │   ├── requests [required: Any, installed: 2.32.5]
│   │   │   ├── charset-normalizer [required: >=2,<4, installed: 3.4.3]
│   │   │   ├── idna [required: >=2.5,<4, installed: 3.10]
│   │   │   ├── urllib3 [required: >=1.21.1,<3, installed: 2.5.0]
│   │   │   └── certifi [required: >=2017.4.17, installed: 2025.8.3]
│   │   ├── tqdm [required: >=4.42.1, installed: 4.67.1]
│   │   │   └── colorama [required: Any, installed: 0.4.6]
│   │   └── typing_extensions [required: >=3.7.4.3, installed: 4.12.2]
│   ├── numpy [required: >=1.17, installed: 1.23.5]
│   ├── packaging [required: >=20.0, installed: 25.0]
│   ├── PyYAML [required: >=5.1, installed: 6.0.2]
│   ├── regex [required: !=2019.12.17, installed: 2025.9.1]
│   ├── requests [required: Any, installed: 2.32.5]
│   │   ├── charset-normalizer [required: >=2,<4, installed: 3.4.3]
│   │   ├── idna [required: >=2.5,<4, installed: 3.10]
│   │   ├── urllib3 [required: >=1.21.1,<3, installed: 2.5.0]
│   │   └── certifi [required: >=2017.4.17, installed: 2025.8.3]
│   ├── tokenizers [required: >=0.22.0,<=0.23.0, installed: 0.22.0]
│   │   └── huggingface-hub [required: >=0.16.4,<1.0, installed: 0.34.4]
│   │       ├── filelock [required: Any, installed: 3.13.1]
│   │       ├── fsspec [required: >=2023.5.0, installed: 2024.6.1]
│   │       ├── packaging [required: >=20.9, installed: 25.0]
│   │       ├── PyYAML [required: >=5.1, installed: 6.0.2]
│   │       ├── requests [required: Any, installed: 2.32.5]
│   │       │   ├── charset-normalizer [required: >=2,<4, installed: 3.4.3]
│   │       │   ├── idna [required: >=2.5,<4, installed: 3.10]
│   │       │   ├── urllib3 [required: >=1.21.1,<3, installed: 2.5.0]
│   │       │   └── certifi [required: >=2017.4.17, installed: 2025.8.3]
│   │       ├── tqdm [required: >=4.42.1, installed: 4.67.1]
│   │       │   └── colorama [required: Any, installed: 0.4.6]
│   │       └── typing_extensions [required: >=3.7.4.3, installed: 4.12.2]
│   ├── safetensors [required: >=0.4.3, installed: 0.6.2]
│   └── tqdm [required: >=4.27, installed: 4.67.1]
│       └── colorama [required: Any, installed: 0.4.6]
├── v-diffusion-pytorch [required: ==0.0.2, installed: 0.0.2]
│   ├── ftfy [required: Any, installed: 6.3.1]
│   │   └── wcwidth [required: Any, installed: 0.2.13]
│   ├── regex [required: Any, installed: 2025.9.1]
│   ├── requests [required: Any, installed: 2.32.5]
│   │   ├── charset-normalizer [required: >=2,<4, installed: 3.4.3]
│   │   ├── idna [required: >=2.5,<4, installed: 3.10]
│   │   ├── urllib3 [required: >=1.21.1,<3, installed: 2.5.0]
│   │   └── certifi [required: >=2017.4.17, installed: 2025.8.3]
│   ├── pillow [required: Any, installed: 10.4.0]
│   ├── torch [required: Any, installed: 2.8.0+cu129]
│   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   ├── networkx [required: Any, installed: 3.3]
│   │   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   └── fsspec [required: Any, installed: 2024.6.1]
│   ├── torchvision [required: Any, installed: 0.23.0+cu129]
│   │   ├── numpy [required: Any, installed: 1.23.5]
│   │   ├── torch [required: ==2.8.0+cu129, installed: 2.8.0+cu129]
│   │   │   ├── filelock [required: Any, installed: 3.13.1]
│   │   │   ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│   │   │   ├── sympy [required: >=1.13.3, installed: 1.13.3]
│   │   │   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   │   ├── networkx [required: Any, installed: 3.3]
│   │   │   ├── Jinja2 [required: Any, installed: 3.1.4]
│   │   │   │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│   │   │   └── fsspec [required: Any, installed: 2024.6.1]
│   │   └── pillow [required: >=5.3.0,!=8.3.*, installed: 10.4.0]
│   └── tqdm [required: Any, installed: 4.67.1]
│       └── colorama [required: Any, installed: 0.4.6]
├── vector-quantize-pytorch [required: ==1.14.41, installed: 1.14.41]
│   ├── einops [required: >=0.8.0, installed: 0.8.1]
│   ├── einx [required: >=0.2.2, installed: 0.3.0]
│   │   ├── numpy [required: Any, installed: 1.23.5]
│   │   ├── sympy [required: Any, installed: 1.13.3]
│   │   │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│   │   └── frozendict [required: Any, installed: 2.4.6]
│   └── torch [required: >=2.0, installed: 2.8.0+cu129]
│       ├── filelock [required: Any, installed: 3.13.1]
│       ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
│       ├── sympy [required: >=1.13.3, installed: 1.13.3]
│       │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
│       ├── networkx [required: Any, installed: 3.3]
│       ├── Jinja2 [required: Any, installed: 3.1.4]
│       │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
│       └── fsspec [required: Any, installed: 2024.6.1]
├── wandb [required: ==0.15.4, installed: 0.15.4]
│   ├── click [required: >=7.0,!=8.0.0, installed: 8.2.1]
│   │   └── colorama [required: Any, installed: 0.4.6]
│   ├── GitPython [required: >=1.0.0,!=3.1.29, installed: 3.1.45]
│   │   └── gitdb [required: >=4.0.1,<5, installed: 4.0.12]
│   │       └── smmap [required: >=3.0.1,<6, installed: 5.0.2]
│   ├── requests [required: >=2.0.0,<3, installed: 2.32.5]
│   │   ├── charset-normalizer [required: >=2,<4, installed: 3.4.3]
│   │   ├── idna [required: >=2.5,<4, installed: 3.10]
│   │   ├── urllib3 [required: >=1.21.1,<3, installed: 2.5.0]
│   │   └── certifi [required: >=2017.4.17, installed: 2025.8.3]
│   ├── psutil [required: >=5.0.0, installed: 7.0.0]
│   ├── sentry-sdk [required: >=1.0.0, installed: 2.37.1]
│   │   ├── urllib3 [required: >=1.26.11, installed: 2.5.0]
│   │   └── certifi [required: Any, installed: 2025.8.3]
│   ├── docker-pycreds [required: >=0.4.0, installed: 0.4.0]
│   │   └── six [required: >=1.4.0, installed: 1.17.0]
│   ├── PyYAML [required: Any, installed: 6.0.2]
│   ├── pathtools [required: Any, installed: 0.1.2]
│   ├── setproctitle [required: Any, installed: 1.3.7]
│   ├── setuptools [required: Any, installed: 80.9.0]
│   ├── appdirs [required: >=1.4.3, installed: 1.4.4]
│   └── protobuf [required: >=3.19.0,<5,!=4.21.0, installed: 3.19.6]
└── webdataset [required: ==0.2.100, installed: 0.2.100]
    ├── braceexpand [required: Any, installed: 0.1.7]
    ├── numpy [required: Any, installed: 1.23.5]
    └── PyYAML [required: Any, installed: 6.0.2]
torchdata==0.11.0
├── urllib3 [required: >=1.25, installed: 2.5.0]
├── requests [required: Any, installed: 2.32.5]
│   ├── charset-normalizer [required: >=2,<4, installed: 3.4.3]
│   ├── idna [required: >=2.5,<4, installed: 3.10]
│   ├── urllib3 [required: >=1.21.1,<3, installed: 2.5.0]
│   └── certifi [required: >=2017.4.17, installed: 2025.8.3]
└── torch [required: >=2, installed: 2.8.0+cu129]
    ├── filelock [required: Any, installed: 3.13.1]
    ├── typing_extensions [required: >=4.10.0, installed: 4.12.2]
    ├── sympy [required: >=1.13.3, installed: 1.13.3]
    │   └── mpmath [required: >=1.1.0,<1.4, installed: 1.3.0]
    ├── networkx [required: Any, installed: 3.3]
    ├── Jinja2 [required: Any, installed: 3.1.4]
    │   └── MarkupSafe [required: >=2.0, installed: 2.1.5]
    └── fsspec [required: Any, installed: 2024.6.1]
typing-inspection==0.4.1
└── typing_extensions [required: >=4.12.0, installed: 4.12.2]
wheel==0.45.1

[sskalnik]

Please copy and paste the errors you encounter.