From 0aad6359e22e04afdb2cf811400bfceef3a34a57 Mon Sep 17 00:00:00 2001
From: Xin Chen <irexyc@gmail.com>
Date: Tue, 6 Sep 2022 19:02:45 +0800
Subject: [PATCH] fix inference_model api

---
 mmdeploy/apis/inference.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mmdeploy/apis/inference.py b/mmdeploy/apis/inference.py
index 507fb36ae..3f02caea8 100644
--- a/mmdeploy/apis/inference.py
+++ b/mmdeploy/apis/inference.py
@@ -48,6 +48,6 @@ def inference_model(model_cfg: Union[str, mmengine.Config],
     model_inputs, _ = task_processor.create_input(img, input_shape)
 
     with torch.no_grad():
-        result = model.test_step([model_inputs])
+        result = model.test_step(model_inputs)
 
     return result