export onnx using cpu model

Signed-off-by: ChengZi <chen.zhang@zilliz.com>
2 years ago · 884323d93d
1 changed files with 1 additions and 1 deletions
--- a/rerank.py
+++ b/rerank.py
@ -122,7 +122,7 @@ class ReRank(NNOperator):
            onnx_config = model_onnx_config(self._model.config)
            onnx_inputs, onnx_outputs = export(
                self.tokenizer,
                self._model,
                self._model.to('cpu'),
                config=onnx_config,
                opset=13,
                output=Path(path)