print(f'mslite_infer:\n{outputs[0].get_data_to_numpy()}\n{outputs[0].get_data_to_numpy().shape}')returnoutputs[0].get_data_to_numpy()defmain():pt_path ='xxx/bge-reranker-base'ms_path ='xxx/bge-reranker-base/onnx/model.mindir'device = torch.device('npu:0') pairs = [['...
Breadcrumbs FastGPT /python /bge-rerank / bge-reranker-base/ Directory actions More options Latest commit 4 people 4.7.1-alpha2 (#1153)Apr 8, 2024 1fbc407· Apr 8, 2024 HistoryHistory Folders and files Name Last commit message Last commit date parent directory .. Dockerfile 4.7.1-alpha...