28 lines
425 B
Plaintext
28 lines
425 B
Plaintext
# Core dependencies
|
|
torch>=2.0.0
|
|
transformers>=4.40.0
|
|
accelerate>=0.25.0
|
|
modelscope>=1.11.0
|
|
|
|
# Inference optimization
|
|
vllm>=0.4.0
|
|
auto-gptq>=0.7.0
|
|
optimum>=1.18.0
|
|
|
|
# Benchmarking
|
|
pytest>=7.4.0
|
|
pytest-asyncio>=0.21.0
|
|
psutil>=5.9.0
|
|
py3nvml>=0.2.7
|
|
|
|
# Utilities
|
|
pydantic>=2.0.0
|
|
python-dotenv>=1.0.0
|
|
tqdm>=4.66.0
|
|
pandas>=2.0.0
|
|
matplotlib>=3.7.0
|
|
seaborn>=0.12.0
|
|
|
|
# Optional for accuracy testing
|
|
datasets>=2.14.0
|
|
scikit-learn>=1.3.0 |