diff --git a/examples/models/llama/main.cpp b/examples/models/llama/main.cpp
index b655a619b26..b22b055c507 100644
--- a/examples/models/llama/main.cpp
+++ b/examples/models/llama/main.cpp
@@ -72,6 +72,8 @@ DEFINE_string(
     "etdump.in",
     "If an etdump path is provided, generate an ETDump file at the specified path for profiling purposes.");
 
+std::vector<std::string> parseStringList(const std::string& input);
+
 // Helper function to parse comma-separated string lists
 std::vector<std::string> parseStringList(const std::string& input) {
   std::vector<std::string> result;
diff --git a/examples/models/llama/targets.bzl b/examples/models/llama/targets.bzl
index 42512145eed..940a0ceafdb 100644
--- a/examples/models/llama/targets.bzl
+++ b/examples/models/llama/targets.bzl
@@ -5,8 +5,8 @@ def define_common_targets():
         if not runtime.is_oss or not aten:
             aten_suffix = "_aten" if aten else ""
 
-            runtime.cxx_binary(
-                name = "main" + aten_suffix,
+            runtime.cxx_library(
+                name = "main_lib" + aten_suffix,
                 srcs = [
                     "main.cpp",
                 ],
@@ -26,3 +26,11 @@ def define_common_targets():
                 ],
                 **get_oss_build_kwargs()
             )
+
+            runtime.cxx_binary(
+                name = "main" + aten_suffix,
+                srcs = [],
+                deps = [
+                    ":main_lib" + aten_suffix,
+                ],
+            )