diff --git a/mmdeploy/apis/onnx/export.py b/mmdeploy/apis/onnx/export.py
index 76c4fb1003..790d002509 100644
--- a/mmdeploy/apis/onnx/export.py
+++ b/mmdeploy/apis/onnx/export.py
@@ -132,7 +132,7 @@ def wrapper(*arg, **kwargs):
         if isinstance(args, torch.Tensor):
             args = args.cpu()
         elif isinstance(args, (tuple, list)):
-            args = tuple([_.cpu() for _ in args])
+            args = tuple([_.cpu() if isinstance(_, torch.Tensor) else _ for _ in args])
         else:
             raise RuntimeError(f'Not supported args: {args}')
         torch.onnx.export(