Windows support!!

[d8ahazard]

It builds (CU124), inference on Windows (Zonos) works.

Required install of triton to compile.
https://github.com/woct0rdho/triton-windows/releases/download/v3.2.0-windows.post9/triton-3.2.0-cp310-cp310-win_amd64.whl

Pre-built wheels for anybody who wants to play at home:

https://github.com/d8ahazard/AudioLab/releases/download/1.0.0/causal_conv1d-1.5.0.post8-cp310-cp310-win_amd64.whl
https://github.com/d8ahazard/AudioLab/releases/download/1.0.0/mamba_ssm-2.2.4-cp310-cp310-win_amd64.whl

[loscrossos]

im getting this: :(

File "C:\Zonos\zonos\model.py", line 8, in <module>
    from mamba_ssm.utils.generation import InferenceParams
  File "C:\Zonos\.env_windows\lib\site-packages\mamba_ssm\__init__.py", line 3, in <module>
    from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
  File "C:\Zonos\.env_windows\lib\site-packages\mamba_ssm\ops\selective_scan_interface.py", line 18, in <module>
    import selective_scan_cuda
ImportError: DLL load failed while importing selective_scan_cuda: The specified procedure could not be found.

could you post your zonos project setup?

edit: FIXED:

i had pytorch 2.6.0 installed. i downgraded and now its working!

torch==2.5.1
torchaudio==2.5.1

[SPOOKEXE]

fireball 🔥

[DarkAlchy]

Need those for Python 3.11 if possible.