numpy
pillow
torch
opencv-python
matplotlib
jupyter
matplotlib
cmapy
torchvision
xformers
accelerate
