JDSS/mem0
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(huggingface): use self.config instead of raw config parameter (#5538)

Browse Source
This commit is contained in:
Hrushikesh Yadav
2026-06-16 15:51:27 +05:30
committed by GitHub
parent 8b370def80
commit 6a6dfb4935
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mem0/embeddings/huggingface.py
+2 -2
View File
@@ -16,8 +16,8 @@ class HuggingFaceEmbedding(EmbeddingBase):
def __init__(self, config: Optional[BaseEmbedderConfig] = None): def __init__(self, config: Optional[BaseEmbedderConfig] = None):
super().__init__(config) super().__init__(config)
if config.huggingface_base_url: if self.config.huggingface_base_url:
self.client = OpenAI(base_url=config.huggingface_base_url) self.client = OpenAI(base_url=self.config.huggingface_base_url)
self.config.model = self.config.model or "tei" self.config.model = self.config.model or "tei"
else: else:
self.config.model = self.config.model or "multi-qa-MiniLM-L6-cos-v1" self.config.model = self.config.model or "multi-qa-MiniLM-L6-cos-v1"