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

Merge pull request #14 from soulteary/fix/user-warning-typedstorage

Browse Source 
fix: ignore typedstorage deprecated message #13
This commit is contained in:
ywfang 2024-02-04 11:54:50 +08:00 committed by GitHub
commit 6eec0fa36b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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, AutoTokenizer,
TextIteratorStreamer TextIteratorStreamer
) )
import warnings
warnings.filterwarnings('ignore', category=UserWarning, message='TypedStorage is deprecated')
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--model_path", type=str, default="") parser.add_argument("--model_path", type=str, default="")