diff --git a/extension/llm/cache/BUCK b/extension/llm/cache/BUCK
index ea782dafc23..ab6da8818c8 100644
--- a/extension/llm/cache/BUCK
+++ b/extension/llm/cache/BUCK
@@ -17,6 +17,19 @@ load(":targets.bzl", "define_common_targets")
 
 fbcode_target(_kind = define_common_targets,)
 
+fbcode_target(_kind = runtime.python_library,
+    name = "cache",
+    srcs = [
+        "reference_cache.py",
+        "update_and_attend.py",
+    ],
+    visibility = ["PUBLIC"],
+    deps = [
+        "//caffe2:torch",
+        "//executorch/exir:_warnings",
+    ],
+)
+
 fbcode_target(_kind = runtime.python_test,
     name = "test_update_and_attend",
     srcs = [
diff --git a/extension/llm/cache/targets.bzl b/extension/llm/cache/targets.bzl
index a362e2cda77..a50cfa6455e 100644
--- a/extension/llm/cache/targets.bzl
+++ b/extension/llm/cache/targets.bzl
@@ -1,15 +1,2 @@
-load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
-
-
 def define_common_targets():
-    runtime.python_library(
-        name = "cache",
-        srcs = [
-            "reference_cache.py",
-            "update_and_attend.py",
-        ],
-        visibility = ["PUBLIC"],
-        deps = [
-            "//caffe2:torch",
-        ],
-    )
+    pass