torch
torchvision
salesforce-lavis