From 74b1b9fc69604d6a43b0ded2b4668c435c2c80f8 Mon Sep 17 00:00:00 2001 From: Christine Tung Date: Mon, 13 Jul 2026 15:45:31 -0700 Subject: [PATCH] No public description PiperOrigin-RevId: 947285116 --- .../checkpoint/experimental/model_surgery/transformations/BUILD | 2 ++ 1 file changed, 2 insertions(+) diff --git a/checkpoint/orbax/checkpoint/experimental/model_surgery/transformations/BUILD b/checkpoint/orbax/checkpoint/experimental/model_surgery/transformations/BUILD index 2f2c7210d..63520ad55 100644 --- a/checkpoint/orbax/checkpoint/experimental/model_surgery/transformations/BUILD +++ b/checkpoint/orbax/checkpoint/experimental/model_surgery/transformations/BUILD @@ -6,6 +6,7 @@ py_library( name = "renaming", srcs = ["renaming.py"], visibility = [ + "//third_party/py/qwix/google/offline_quantization_example:__pkg__", "//third_party/py/qwix/integration_tests:__pkg__", "//third_party/py/tpu_kernel_testbench:__pkg__", ], @@ -36,6 +37,7 @@ py_library( name = "nesting", srcs = ["nesting.py"], visibility = [ + "//third_party/py/qwix/google/offline_quantization_example:__pkg__", "//third_party/py/qwix/integration_tests:__pkg__", "//third_party/py/tpu_kernel_testbench:__pkg__", ],