RYDE-WORK/MiniCPM
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/MiniCPM.git synced 2026-01-19 21:03:39 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: ignore typedstorage deprecated message

Browse Source
This commit is contained in:
Su Yang 2024-02-02 14:02:20 +08:00
parent 7333ec793b
commit 5c43592042
No known key found for this signature in database
GPG Key ID: DBCDD8CBF440F8DE
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
demo/hf_based_demo.py
View File

@ -11,6 +11,8 @@ from transformers import (
AutoTokenizer,
TextIteratorStreamer
)
import warnings
warnings.filterwarnings('ignore', category=UserWarning, message='TypedStorage is deprecated')
parser = argparse.ArgumentParser()
parser.add_argument("--model_path", type=str, default="")