numpy<2.0
albumentations<2.0
timm
torch
pillow