torch
pytorch_lightning
transformers
tqdm
sentence_transformers
clip
torchmetrics
