diff --git a/ACL_PyTorch/built-in/cv/GroundingDINO/video_demo_npu.py b/ACL_PyTorch/built-in/cv/GroundingDINO/video_demo_npu.py index ac7b012c0e1b55be70aac7dd5795fec3b81c2bc6..aa081657c35395a2fd26bb1ae0b8008682040706 100644 --- a/ACL_PyTorch/built-in/cv/GroundingDINO/video_demo_npu.py +++ b/ACL_PyTorch/built-in/cv/GroundingDINO/video_demo_npu.py @@ -119,7 +119,7 @@ def preprocess(model, batch_frame, text_prompt): batch_data['inputs'].append(data_['inputs']) batch_data['data_samples'].append(data_['data_samples']) - return batch_data + return batch_data def infer(model, batch_data, tokenized, attention_mask, position_ids):