| # Base dependencies | |
| torch>=2.0.0 | |
| transformers>=4.34.0 | |
| huggingface_hub>=0.17.1 | |
| peft>=0.5.0 | |
| trl>=0.7.4 | |
| datasets>=2.14.0 | |
| llama-cpp-python>=0.2.0 | |
| fastapi>=0.103.2 | |
| uvicorn>=0.23.2 | |
| pydantic>=2.4.2 | |
| numpy>=1.24.0 | |
| # Training dependencies | |
| accelerate>=0.23.0 | |
| bitsandbytes>=0.41.0 | |
| # Utils | |
| requests>=2.31.0 | |
| tqdm>=4.66.1 |