Spaces:
Running
Running
| # Requirements ROCM | |
| # Core dependencies | |
| transformers>=4.51.0,<4.58.0 | |
| diffusers | |
| gradio==6.2.0 | |
| matplotlib>=3.7.5 | |
| scipy>=1.10.1 | |
| soundfile>=0.13.1 | |
| loguru>=0.7.3 | |
| einops>=0.8.1 | |
| accelerate>=1.12.0 | |
| fastapi>=0.110.0 | |
| uvicorn[standard]>=0.27.0 | |
| numba>=0.63.1 | |
| vector-quantize-pytorch>=1.27.15 | |
| # torchcodec>=0.9.1 # Not supported on ROCM - soundfile fallback will be used | |
| torchao | |
| toml | |
| modelscope | |
| # Training dependencies (required for LoRA training) | |
| peft>=0.18.0 | |
| lightning>=2.0.0 | |
| tensorboard>=2.0.0 | |
| xxhash | |
| # Local package - install with: pip install -e acestep/third_parts/nano-vllm | |
| # nano-vllm | |