zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-30 02:53:41 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

remove transformer flax/torch warning

Browse Source
This commit is contained in:
Awni Hannun 2024-10-26 18:11:14 -07:00
parent 00b67e2f97
commit db7d3c7a76
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
llms/mlx_lm/__init__.py
View File

@ -1,4 +1,10 @@
# Copyright © 2023-2024 Apple Inc. # Copyright © 2023-2024 Apple Inc.
from ._version import __version__ from ._version import __version__
import os
os.environ["TRANSFORMERS_NO_ADVISORY_WARNINGS"] = "1"
from .utils import convert, generate, load, stream_generate from .utils import convert, generate, load, stream_generate