From db7d3c7a7618cc99f84f06f19c71e55d6ec59033 Mon Sep 17 00:00:00 2001 From: Awni Hannun Date: Sat, 26 Oct 2024 18:11:14 -0700 Subject: [PATCH] remove transformer flax/torch warning --- llms/mlx_lm/__init__.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/llms/mlx_lm/__init__.py b/llms/mlx_lm/__init__.py index 502c78e5..e223d6d4 100644 --- a/llms/mlx_lm/__init__.py +++ b/llms/mlx_lm/__init__.py @@ -1,4 +1,10 @@ # Copyright © 2023-2024 Apple Inc. from ._version import __version__ + +import os +os.environ["TRANSFORMERS_NO_ADVISORY_WARNINGS"] = "1" + from .utils import convert, generate, load, stream_generate + +