From ff6b251c66caa6ba42494dc2397c6a465fda31bd Mon Sep 17 00:00:00 2001 From: Dev Khant Date: Thu, 10 Apr 2025 01:01:00 +0530 Subject: [PATCH] Handle HF logging (#2523) --- mem0/embeddings/huggingface.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/mem0/embeddings/huggingface.py b/mem0/embeddings/huggingface.py index 913d5882..6e77d6b3 100644 --- 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