
[all]
torch>=2.0
transformers>=4.30
pynvml>=11.5
psutil>=5.9
numpy>=1.24
openai>=1.0

[dev]
pytest>=7.0
pytest-asyncio>=0.21
build>=0.10
twine>=4.0

[gpu]
pynvml>=11.5
psutil>=5.9

[openai]
openai>=1.0

[torch]
torch>=2.0

[transformers]
transformers>=4.30
torch>=2.0
