fix inference_model api

open-mmlab · Sep 6, 2022 · 0aad635 · 0aad635
1 parent a2f8287
commit 0aad635
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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