| # core deps | |
| transformers>=4.49.0,<=4.56.2,!=4.52.0; python_version < '3.10' | |
| transformers>=4.49.0,<=4.57.1,!=4.52.0,!=4.57.0; python_version >= '3.10' | |
| datasets>=2.16.0,<=4.0.0 | |
| accelerate>=1.3.0,<=1.11.0 | |
| peft>=0.14.0,<=0.17.1 | |
| trl>=0.8.6,<=0.9.6 | |
| # gui | |
| gradio>=4.38.0,<=5.45.0 | |
| matplotlib>=3.7.0 | |
| tyro<0.9.0 | |
| # ops | |
| einops | |
| numpy<2.0.0 | |
| pandas>=2.0.0 | |
| scipy | |
| # model and tokenizer | |
| sentencepiece | |
| tiktoken | |
| modelscope>=1.14.0 | |
| hf-transfer | |
| safetensors<=0.5.3 | |
| # python | |
| fire | |
| omegaconf | |
| packaging | |
| protobuf | |
| pyyaml | |
| pydantic<=2.10.6 | |
| # api | |
| uvicorn | |
| fastapi | |
| sse-starlette | |
| # media | |
| av | |
| librosa | |
| # yanked | |
| propcache!=0.4.0 | |