Update requirements.txt

requirements.txt

@@ -1,328 +1,131 @@
-# This file was autogenerated by uv via the following command:
-#    uv pip compile pyproject.toml -o requirements.txt
-aiofiles==23.2.1
-    # via gradio
-annotated-types==0.7.0
-    # via pydantic
-anyio==4.8.0
-    # via
-    #   gradio
-    #   httpx
-    #   starlette
-attrs==25.1.0
-    # via
-    #   clldutils
-    #   csvw
-    #   jsonschema
-    #   phonemizer
-    #   referencing
-babel==2.17.0
-    # via csvw
-certifi==2025.1.31
-    # via
-    #   httpcore
-    #   httpx
-    #   requests
-cffi==1.17.1
-    # via soundfile
-charset-normalizer==3.4.1
-    # via requests
-click==8.1.8
-    # via
-    #   typer
-    #   uvicorn
-clldutils==3.21.0
-    # via segments
-colorama==0.4.6
-    # via csvw
-colorlog==6.9.0
-    # via clldutils
-csvw==3.5.1
-    # via segments
-dlinfo==2.0.0
-    # via phonemizer
-exceptiongroup==1.2.2
-    # via anyio
+# Core AI / ML dependencies
+torch==2.4.0
+torchaudio==2.4.0
+triton==3.0.0
+flash-attn>=2.5.5  # Needed for rotary embedding with torch 2.4.0
+
+# Optional: Mamba SSM (uncomment if you're using Mamba-based models)
+# git+https://github.com/state-spaces/mamba.git
+
+# Web framework and API
 fastapi==0.115.8
-    # via gradio
-ffmpy==0.5.0
-    # via gradio
-filelock==3.17.0
-    # via
-    #   huggingface-hub
-    #   torch
-    #   transformers
-    #   triton
-fsspec==2025.2.0
-    # via
-    #   gradio-client
-    #   huggingface-hub
-    #   torch
-gradio==5.16.0
-    # via
-    #   zonos (pyproject.toml)
-    #   spaces
-gradio-client==1.7.0
-    # via gradio
-h11==0.14.0
-    # via
-    #   httpcore
-    #   uvicorn
-hf-transfer==0.1.9
-    # via zonos (pyproject.toml)
-httpcore==1.0.7
-    # via httpx
-httpx==0.28.1
-    # via
-    #   gradio
-    #   gradio-client
-    #   safehttpx
-    #   spaces
+uvicorn==0.34.0
+starlette==0.45.3
+python-multipart==0.0.20
+
+# Audio and file handling
+soundfile==0.13.1
+pydub==0.25.1
+aiofiles==23.2.1
+
+# NLP, transformers
+transformers==4.48.3
+tokenizers==0.21.0
+phonemizer==3.3.0
+safetensors==0.5.2
+
+# Hugging Face ecosystem
 huggingface-hub==0.28.1
-    # via
-    #   zonos (pyproject.toml)
-    #   gradio
-    #   gradio-client
-    #   tokenizers
-    #   transformers
-idna==3.10
-    # via
-    #   anyio
-    #   httpx
-    #   requests
-inflect==7.5.0
-    # via zonos (pyproject.toml)
-isodate==0.7.2
-    # via
-    #   csvw
-    #   rdflib
-jinja2==3.1.5
-    # via
-    #   gradio
-    #   torch
-joblib==1.4.2
-    # via phonemizer
-jsonschema==4.23.0
-    # via csvw
-jsonschema-specifications==2024.10.1
-    # via jsonschema
-kanjize==1.6.0
-    # via zonos (pyproject.toml)
-language-tags==1.2.0
-    # via csvw
-lxml==5.3.1
-    # via clldutils
-markdown==3.7
-    # via clldutils
-markdown-it-py==3.0.0
-    # via rich
-markupsafe==2.1.5
-    # via
-    #   clldutils
-    #   gradio
-    #   jinja2
-mdurl==0.1.2
-    # via markdown-it-py
-more-itertools==10.6.0
-    # via inflect
-mpmath==1.3.0
-    # via sympy
-networkx==3.4.2
-    # via torch
+hf-transfer==0.1.9
+
+# Numerical + scientific
 numpy==2.2.2
-    # via
-    #   zonos (pyproject.toml)
-    #   gradio
-    #   pandas
-    #   soundfile
-    #   transformers
+pandas==2.2.3
+sympy==1.13.3
+mpmath==1.3.0
+
+# Misc AI + CUDA dependencies
 nvidia-cublas-cu12==12.1.3.1
-    # via
-    #   nvidia-cudnn-cu12
-    #   nvidia-cusolver-cu12
-    #   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
-    #   torch
 nvidia-nccl-cu12==2.20.5
-    # via torch
 nvidia-nvjitlink-cu12==12.8.61
-    # via
-    #   nvidia-cusolver-cu12
-    #   nvidia-cusparse-cu12
 nvidia-nvtx-cu12==12.1.105
-    # via torch
+
+# Gradio (UI) and frontend
+gradio==5.16.0
+gradio-client==1.7.0
 orjson==3.10.15
-    # via gradio
-packaging==24.2
-    # via
-    #   zonos (pyproject.toml)
-    #   gradio
-    #   gradio-client
-    #   huggingface-hub
-    #   spaces
-    #   transformers
-pandas==2.2.3
-    # via gradio
-phonemizer==3.3.0
-    # via zonos (pyproject.toml)
+ruff==0.9.6
+safehttpx==0.1.6
+jinja2==3.1.5
+markdown==3.7
 pillow==11.1.0
-    # via gradio
-psutil==5.9.8
-    # via spaces
-pycparser==2.22
-    # via cffi
+markdown-it-py==3.0.0
+mdurl==0.1.2
+
+# Pydantic & validation
 pydantic==2.10.6
-    # via
-    #   fastapi
-    #   gradio
-    #   spaces
 pydantic-core==2.27.2
-    # via pydantic
-pydub==0.25.1
-    # via gradio
-pygments==2.19.1
-    # via rich
-pylatexenc==2.10
-    # via clldutils
-pyparsing==3.2.1
-    # via rdflib
-python-dateutil==2.9.0.post0
-    # via
-    #   clldutils
-    #   csvw
-    #   pandas
-python-multipart==0.0.20
-    # via gradio
-pytz==2025.1
-    # via pandas
-pyyaml==6.0.2
-    # via
-    #   gradio
-    #   huggingface-hub
-    #   transformers
-rdflib==7.1.3
-    # via csvw
+typing-extensions==4.12.2
+annotated-types==0.7.0
+
+# JSON / RDF / CSV tools
+jsonschema==4.23.0
+jsonschema-specifications==2024.10.1
 referencing==0.36.2
-    # via
-    #   jsonschema
-    #   jsonschema-specifications
-regex==2024.11.6
-    # via
-    #   segments
-    #   transformers
+rpds-py==0.22.3
+rdflib==7.1.3
+csvw==3.5.1
+clldutils==3.21.0
+
+# Request/HTTP libraries
+httpx==0.28.1
+httpcore==1.0.7
 requests==2.32.3
-    # via
-    #   csvw
-    #   huggingface-hub
-    #   spaces
-    #   transformers
+urllib3==2.3.0
+idna==3.10
+certifi==2025.1.31
 rfc3986==1.5.0
-    # via csvw
-rich==13.9.4
-    # via typer
-rpds-py==0.22.3
-    # via
-    #   jsonschema
-    #   referencing
-ruff==0.9.6
-    # via gradio
-safehttpx==0.1.6
-    # via gradio
-safetensors==0.5.2
-    # via transformers
-segments==2.2.1
-    # via phonemizer
-semantic-version==2.10.0
-    # via gradio
-setuptools==75.8.0
-    # via zonos (pyproject.toml)
+charset-normalizer==3.4.1
+
+# Utility
+attrs==25.1.0
+filelock==3.17.0
+packaging==24.2
+regex==2024.11.6
+colorlog==6.9.0
+colorama==0.4.6
+tabulate==0.9.0
 shellingham==1.5.4
-    # via typer
-six==1.17.0
-    # via python-dateutil
+pyyaml==6.0.2
+click==8.1.8
+setuptools==75.8.0
+
+# Rich logging / terminal UI
+rich==13.9.4
+pygments==2.19.1
+
+# AnyIO and async stack
+anyio==4.8.0
 sniffio==1.3.1
-    # via anyio
-soundfile==0.13.1
-    # via zonos (pyproject.toml)
-spaces==0.32.0
-    # via zonos (pyproject.toml)
-starlette==0.45.3
-    # via
-    #   fastapi
-    #   gradio
+exceptiongroup==1.2.2
+
+# Time and date
+python-dateutil==2.9.0.post0
+pytz==2025.1
+tzdata==2025.1
+isodate==0.7.2
+
+# C-extensions
+cffi==1.17.1
+pycparser==2.22
+
+# Extra dependencies from your original list
+inflect==7.5.0
+kanjize==1.6.0
+segments==2.2.1
+language-tags==1.2.0
+dlinfo==2.0.0
+more-itertools==10.6.0
+typeguard==4.4.1
 sudachidict-full==20250129
-    # via zonos (pyproject.toml)
 sudachipy==0.6.10
-    # via
-    #   zonos (pyproject.toml)
-    #   sudachidict-full
-sympy==1.13.3
-    # via torch
-tabulate==0.9.0
-    # via clldutils
-tokenizers==0.21.0
-    # via transformers
-tomlkit==0.13.2
-    # via gradio
-torch==2.4.0
-    # via
-    #   zonos (pyproject.toml)
-    #   torchaudio
-torchaudio==2.4.0
-    # via zonos (pyproject.toml)
-tqdm==4.67.1
-    # via
-    #   huggingface-hub
-    #   transformers
-transformers==4.48.3
-    # via zonos (pyproject.toml)
-triton==3.0.0
-    # via torch
-typeguard==4.4.1
-    # via inflect
-typer==0.15.1
-    # via gradio
-typing-extensions==4.12.2
-    # via
-    #   anyio
-    #   fastapi
-    #   gradio
-    #   gradio-client
-    #   huggingface-hub
-    #   phonemizer
-    #   pydantic
-    #   pydantic-core
-    #   referencing
-    #   rich
-    #   spaces
-    #   torch
-    #   typeguard
-    #   typer
-    #   uvicorn
-tzdata==2025.1
-    # via pandas
 uritemplate==4.1.1
-    # via csvw
-urllib3==2.3.0
-    # via requests
-uvicorn==0.34.0
-    # via gradio
-websockets==14.2
-    # via gradio-client
-
+tomlkit==0.13.2