No public description

PiperOrigin-RevId: 601905126
google-ai-edge · Jan 27, 2024 · 233ecc8 · 233ecc8
1 parent b004a3e
commit 233ecc8
Show file tree

Hide file tree

Showing 9 changed files with 2 additions and 698 deletions.
diff --git a/mediapipe/tasks/tools/BUILD → mediapipe/model_maker/python/llm/BUILD b/mediapipe/tasks/tools/BUILD → mediapipe/model_maker/python/llm/BUILD
diff --git a/mediapipe/tasks/tools/converter_base.py → .../model_maker/python/llm/converter_base.py b/mediapipe/tasks/tools/converter_base.py → .../model_maker/python/llm/converter_base.py
diff --git a/mediapipe/tasks/tools/quantization_util.py → ...del_maker/python/llm/quantization_util.py b/mediapipe/tasks/tools/quantization_util.py → ...del_maker/python/llm/quantization_util.py
diff --git a/...ipe/tasks/tools/quantization_util_test.py → ...aker/python/llm/quantization_util_test.py b/...ipe/tasks/tools/quantization_util_test.py → ...aker/python/llm/quantization_util_test.py
@@ -19,7 +19,7 @@
 from jax import numpy as jnp
 import numpy as np
 
-from mediapipe.tasks.tools import quantization_util
+from mediapipe.model_maker.python.llm import quantization_util
 
 
 _dtype = lambda x: getattr(x, 'dtype', None) or np.asarray(x).dtype

diff --git a/mediapipe/tasks/testdata/text/BUILD b/mediapipe/tasks/testdata/text/BUILD
@@ -20,6 +20,7 @@ load(
 package(
     default_visibility = [
         "//mediapipe/calculators/tensor:__subpackages__",
+        "//mediapipe/model_maker:__subpackages__",
         "//mediapipe/tasks:__subpackages__",
     ],
     licenses = ["notice"],  # Apache 2.0

diff --git a/mediapipe/tasks/tools/pytorch_converter.py b/mediapipe/tasks/tools/pytorch_converter.py
diff --git a/mediapipe/tasks/tools/pytorch_converter_test.py b/mediapipe/tasks/tools/pytorch_converter_test.py