Pillow torch transformers==4.46.1 numpy timm einops