safetensors
torch
tqdm
numpy