torch==2.11.0 torchvision transformers==5.3.0 accelerate tokenizers sentencepiece timm einops numpy==1.26.4 pillow matplotlib opencv-python-headless albumentations megfile orjson loguru