from sentence_transformers import SentenceTransformer

if __name__ == '__main__':
    model = SentenceTransformer('E:\\Data\\Research\\Models\\tapex-base-finetuned-wikisql')
    sentences = ['公积金转入深圳', '公积金转出深圳']
    embedding = model.encode(sentences, convert_to_tensor=True, device="cuda")
    print(1)