klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Handle HF logging (#2523)

Browse Source
This commit is contained in:
Dev Khant
2025-04-10 01:01:00 +05:30
committed by GitHub
parent 55df395fd6
commit ff6b251c66
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
mem0/embeddings/huggingface.py
View File

@@ -1,5 +1,10 @@
import logging
from typing import Literal, Optional from typing import Literal, Optional
logging.getLogger("transformers").setLevel(logging.WARNING)
logging.getLogger("sentence_transformers").setLevel(logging.WARNING)
logging.getLogger("huggingface_hub").setLevel(logging.WARNING)
from sentence_transformers import SentenceTransformer from sentence_transformers import SentenceTransformer
from mem0.configs.embeddings.base import BaseEmbedderConfig from mem0.configs.embeddings.base import BaseEmbedderConfig