datasets
transformers
torch
elasticsearch==8.18.1
aiohttp
ujson
scipy
faiss-gpu-cu12