Add image embedding task and test case

This commit is contained in:
liyaxing
2023-09-27 17:24:38 +08:00
committed by tsukasa
parent 7f3687fab1
commit 7f81e25e3b
-1
View File
@@ -67,7 +67,6 @@ def test_st():
] ]
# Compute embeddings # Compute embeddings
#embeddings = model.encode(sentences, convert_to_tensor=True) #embeddings = model.encode(sentences, convert_to_tensor=True)
embeddings = model.encode(sentences) embeddings = model.encode(sentences)
print("embeddings as follows: ") print("embeddings as follows: ")