numpy
pillow
torch
opencv-python
matplotlib
jupyter
matplotlib
cmapy
torchvision
xformers
accelerate
opt_einsum
timm==0.6.5
flash-attn
albumentations
