torch torchvision torchaudio transformers==4.46.3 tokenizers==0.20.3 PyMuPDF img2pdf einops easydict addict Pillow numpy flash-attn==2.7.3