Handle HF logging (#2523)

2025-04-10 01:01:00 +05:30
parent 55df395fd6
commit ff6b251c66
1 changed files with 5 additions and 0 deletions
--- a/mem0/embeddings/huggingface.py
+++ b/mem0/embeddings/huggingface.py
@@ -1,5 +1,10 @@
+import logging
 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 mem0.configs.embeddings.base import BaseEmbedderConfig