diff --git a/demo/hf_based_demo.py b/demo/hf_based_demo.py index 12ff697..d32c613 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="")