Transfer to cpu only when torch.Tensor arguments in onnx export

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(