torch>=2.0.1
numpy
safetensors>=0.3.1
tqdm
