sentence-transformers numpy torch tqdm