You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
8 lines
333 B
8 lines
333 B
1 year ago
|
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)
|