From 5c435920422a7c2aaab9d68acd9a86c21968e4b0 Mon Sep 17 00:00:00 2001 From: Su Yang Date: Fri, 2 Feb 2024 14:02:20 +0800 Subject: [PATCH] fix: ignore typedstorage deprecated message --- demo/hf_based_demo.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/demo/hf_based_demo.py b/demo/hf_based_demo.py index 1cd9289..863ce69 100644 --- a/demo/hf_based_demo.py +++ b/demo/hf_based_demo.py @@ -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="")