From 7e3340a3fccd002838b83eba083b29c8b2693abc Mon Sep 17 00:00:00 2001 From: Peng Lu Date: Tue, 26 Mar 2024 14:50:20 +0800 Subject: [PATCH] [Fix] fix bug in inferencer demo (#2966) --- demo/inferencer_demo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/demo/inferencer_demo.py b/demo/inferencer_demo.py index d20c433f4e..6aac68c9bc 100644 --- a/demo/inferencer_demo.py +++ b/demo/inferencer_demo.py @@ -83,7 +83,7 @@ def parse_args(): # selected pose2d model args, _ = parser.parse_known_args() for model in POSE2D_SPECIFIC_ARGS: - if model in args.pose2d: + if args.pose2d is not None and model in args.pose2d: filter_args.update(POSE2D_SPECIFIC_ARGS[model]) break