Move logging

This commit is contained in:
Daniel Han 2025-09-03 20:07:27 -07:00
commit 29d2f5f09b
3 changed files with 13 additions and 9 deletions

View file

@ -124,6 +124,8 @@ from .import_fixes import fix_xformers_performance_issue
fix_xformers_performance_issue(); del fix_xformers_performance_issue;
from .import_fixes import fix_vllm_aimv2_issue
fix_vllm_aimv2_issue(); del fix_vllm_aimv2_issue;
from .import_fixes import ignore_logger_messages
ignore_logger_messages(); del ignore_logger_messages;
# Torch 2.4 has including_emulation
if DEVICE_TYPE == "cuda":

View file

@ -117,3 +117,13 @@ def fix_vllm_aimv2_issue():
if UNSLOTH_ENABLE_LOGGING:
print(f"Unsloth: Failed patching vLLM with error = {str(e)}")
pass
def ignore_logger_messages():
# Ignore Environment variable `HF_TOKEN` is set
try:
from huggingface_hub._login.utils import logger as huggingface_hub_logger
huggingface_hub_logger.addFilter(HideLoggingMessage("`HF_TOKEN`"))
del huggingface_hub_logger
except:
pass
pass

View file

@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.
__version__ = "2025.8.10"
__version__ = "2025.9.1"
__all__ = [
"SUPPORTS_BFLOAT16",
@ -317,14 +317,6 @@ try:
except:
pass
# Environment variable `HF_TOKEN` is set
try:
from huggingface_hub._login.utils import logger as huggingface_hub_logger
huggingface_hub_logger.addFilter(HideLoggingMessage("`HF_TOKEN`"))
del huggingface_hub_logger
except:
pass
# Errors out on
# Some weights of Gemma3nForConditionalGeneration were not initialized from the model checkpoint
from transformers.modeling_utils import logger as transformers_logger