Dramabox / requirements.txt
Manmay's picture
Drop mamba-ssm / causal-conv1d from requirements
e44cca0
torch==2.8.0
torchaudio==2.8.0
# pydantic 2.11+ emits bool-shorthand `additionalProperties: True` which crashes
# gradio_client's get_type (`if "const" in schema`). 2.10.6 is the last version
# that still emits the dict form. Tracked at:
# https://discuss.huggingface.co/t/154601
pydantic==2.11.10
safetensors>=0.4.0
accelerate>=0.25.0
peft>=0.7.0
# xformers removed: 0.0.32.post2's bundled FA3 Hopper kernel raises
# "CUDA error: invalid argument" on ZeroGPU's H200 runtime. attention.py
# falls back to torch SDPA cleanly when xformers is absent (try/except
# import at module level), and SDPA on torch 2.8 has its own efficient
# H200 backends.
av>=12.0.0
einops>=0.7.0
PyYAML>=6.0
sentencepiece>=0.1.99
transformers>=4.45.0
huggingface_hub>=0.20.0,<1.0
bitsandbytes>=0.45.0
gradio==6.14.0
spaces>=0.30.0
soundfile>=0.12.0
resemble-perth @ git+https://github.com/resemble-ai/Perth.git@master
# ── RE-USE deps intentionally omitted on HF Spaces ──────────────────────────
# The mamba-ssm / causal-conv1d wheels need nvcc + matching CUDA toolkit at
# install time, which HF Spaces' build container doesn't provide. Without
# them, super_resolution.REUSEUpsampler raises ImportError on first use and
# TTSServer._denoise_voice_ref logs a one-time warning + silently skips the
# voice-ref denoise for the rest of the session. Long-form chunking + the
# rest of the model are unaffected. To enable denoise_ref on the Space,
# switch to a Dockerfile-based Space build with nvcc available, then add:
# resampy>=0.4.0
# mamba-ssm>=2.2.0 ; platform_system == "Linux"
# causal-conv1d>=1.4.0 ; platform_system == "Linux"