This commit is contained in:
parent
8b15c7553c
commit
467f85fc1b
|
|
@ -20,6 +20,8 @@ USER_DOC_PATH = "docs"
|
|||
|
||||
|
||||
class CustomEmbedding(BaseEmbedding):
|
||||
model: SentenceTransformer # ✅ 显式声明是字段
|
||||
|
||||
def __init__(self, model_name: str):
|
||||
from sentence_transformers import SentenceTransformer
|
||||
self.model = SentenceTransformer(model_name)
|
||||
|
|
|
|||
Loading…
Reference in New Issue