diff --git a/src/main.cpp b/src/main.cpp
index 4cb011e4dacd253c8512770c711017bf95fa321e..29158de88e115f9a98ef42f24cfd36ec9c455bcf 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -57,6 +57,7 @@ PYBIND11_MODULE(_core, m)
 		.def("getTorchScript", &Model::getTorchScript)
 		.def("getCompiledFunctionName", &Model::getCompiledFunctionName)
 		.def("getFlatParameters", &Model::getFlatParameters)
+		.def("getParameterNames", &Model::getParameterNames)
 		.def("getParameterCount", &Model::getParameterCount)
 		.def("getRecommendedParamIndices", &Model::getRecommendedParamIndices)
 		.def("__repr__", &Model::getModelStr);