diff --git a/main/index.mjs b/main/index.mjs
index 4e09894..6298419 100644
--- a/main/index.mjs
+++ b/main/index.mjs
@@ -27243,7 +27243,11 @@ var __webpack_exports__ = {};
 async function main() {
     const sourceDir = _actions_core__WEBPACK_IMPORTED_MODULE_0__.getInput("source-dir");
     const buildDir = _actions_core__WEBPACK_IMPORTED_MODULE_0__.getInput("build-dir");
-    await _actions_exec__WEBPACK_IMPORTED_MODULE_1__.exec("cmake", [sourceDir || ".", "-B", buildDir || "build"]);
+    const configureArgs = [sourceDir || ".", "-B", buildDir || "build"];
+    const generator = _actions_core__WEBPACK_IMPORTED_MODULE_0__.getInput("generator");
+    if (generator)
+        configureArgs.push(...["-G", generator]);
+    await _actions_exec__WEBPACK_IMPORTED_MODULE_1__.exec("cmake", configureArgs);
     _actions_core__WEBPACK_IMPORTED_MODULE_0__.setOutput("build-dir", buildDir || "build");
     const runBuild = _actions_core__WEBPACK_IMPORTED_MODULE_0__.getBooleanInput("run-build");
     const runTest = _actions_core__WEBPACK_IMPORTED_MODULE_0__.getBooleanInput("run-test");
diff --git a/src/main.mts b/src/main.mts
index eb6ccab..38a2002 100644
--- a/src/main.mts
+++ b/src/main.mts
@@ -4,7 +4,13 @@ import exec from "@actions/exec";
 async function main() {
   const sourceDir = core.getInput("source-dir");
   const buildDir = core.getInput("build-dir");
-  await exec.exec("cmake", [sourceDir || ".", "-B", buildDir || "build"]);
+
+  const configureArgs = [sourceDir || ".", "-B", buildDir || "build"];
+
+  const generator = core.getInput("generator");
+  if (generator) configureArgs.push(...["-G", generator]);
+
+  await exec.exec("cmake", configureArgs);
   core.setOutput("build-dir", buildDir || "build");
 
   const runBuild = core.getBooleanInput("run-build");