fix: transformers logging suppressed definitely

Libertai · Dec 5, 2024 · f379d0f · f379d0f
1 parent f8a3514
commit f379d0f
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 135 deletions.
diff --git a/libertai_agents/libertai_agents/__init__.py b/libertai_agents/libertai_agents/__init__.py
@@ -0,0 +1,4 @@
+import logging
+
+# Disables the error about frameworks not installed
+logging.getLogger("transformers").disabled = True
diff --git a/libertai_agents/libertai_agents/models/base.py b/libertai_agents/libertai_agents/models/base.py
@@ -1,4 +1,3 @@
-import logging
 from abc import ABC, abstractmethod
 from typing import Literal
 
@@ -9,9 +8,6 @@
 )
 from libertai_agents.interfaces.tools import Tool
 
-# Disables the error about models not available
-logging.getLogger("transformers").disabled = True
-
 ModelId = Literal[
     "NousResearch/Hermes-3-Llama-3.1-8B", "mistralai/Mistral-Nemo-Instruct-2407"
 ]