torch
transformers
torchvision
cv2