Update memory.py
This commit is contained in:
parent
d7d7fc2a7d
commit
33ee0f9cb5
|
|
@ -25,7 +25,6 @@ class FinancialSituationMemory:
|
|||
else:
|
||||
model = SentenceTransformer("all-MiniLM-L6-v2")
|
||||
embeddings = model.encode(text, convert_to_numpy=True, precision='int8')
|
||||
# 单文本返回一维数组,多文本返回二维数组
|
||||
return embeddings
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue