File tree Expand file tree Collapse file tree 1 file changed +1
-1
lines changed
onnxruntime/core/providers/tensorrt Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Original file line number Diff line number Diff line change @@ -2921,7 +2921,7 @@ common::Status TensorrtExecutionProvider::RefitEngine(std::string onnx_model_fil
29212921 bool refit_from_file = onnx_model_bytestream == nullptr && onnx_model_bytestream_size == 0 ;
29222922 bool refit_with_external_data_bytestream = onnx_external_data_bytestream != nullptr && onnx_external_data_bytestream_size != 0 ;
29232923 bool refit_with_weights_in_memory = in_memory_weights.size () > 0 ;
2924- bool refit_with_external_data = refit_with_weights_in_memory || refit_with_weights_in_memory ;
2924+ bool refit_with_external_data = refit_with_weights_in_memory || refit_with_external_data_bytestream ;
29252925 bool refit_complete = false ;
29262926 std::filesystem::path onnx_model_path{onnx_model_folder_path};
29272927 if (refit_from_file) {
You can’t perform that action at this time.
0 commit comments