PyYAML
cachetools
psutil
py-itree
urllib3
requests
rich

[fastapi]
fastapi
uvicorn
pandas

[full]
fastapi
uvicorn
pandas
torch>=1.13.1
transformer>=4.26.1
protobuf~=3.20.2
importlib-metadata>=6.0.0
salesforce-lavis
pynvml

[gpu]
pynvml

[lavis]
salesforce-lavis

[transformers]
torch>=1.13.1
transformer>=4.26.1
protobuf~=3.20.2
importlib-metadata>=6.0.0
