torch torchvision einops transformers>=4.28.1 tokenizers>=0.13.3 sentencepiece safetensors>=0.4.2 numpy==2.1.3 pyyaml scipy psutil