importlib-metadata
termcolor
cupy-cuda12x
torch>=1.8.1
torchvision>=0.9.1
