mediapipe-nightly 0.10.10.post20240219__cp39-cp39-macosx_11_0_x86_64.whl → 0.10.10.post20240221__cp39-cp39-macosx_11_0_x86_64.whl

mediapipe/__init__.py

@@ -23,4 +23,4 @@ del modules
 del python
 del mediapipe
 del util
-__version__ = '0.10.10-20240219'
+__version__ = '0.10.10-20240221'

mediapipe/model_maker/__init__.py

@@ -22,7 +22,6 @@ from mediapipe.model_maker.python.vision import gesture_recognizer
 from mediapipe.model_maker.python.text import text_classifier
 from mediapipe.model_maker.python.vision import object_detector
 from mediapipe.model_maker.python.vision import face_stylizer
-from mediapipe.model_maker.python.llm import llm_converter
 
 # Remove duplicated and non-public API
 del python

mediapipe/tasks/python/genai/converter/llm_converter.py

@@ -107,6 +107,7 @@ def quantize_by_actions(
         )
         output_tensors[action.target_name] = (target_var, pack)
         output_tensors[action.target_name + '_quantized_scale'] = (scale, False)
+        zp = None
       else:
         target_var, scale, zp = quantization_util.quantize_tensor(
             var=action.tensor_value,
@@ -114,12 +115,13 @@ def quantize_by_actions(
             sym=is_symmetric,
             number_bits=action.quantize_bits,
         )
-        if backend == 'cpu' and (action.quantize_bits == 4):
-          target_var, scale, zp = quantization_util.update_to_uint4(
-              target_var, scale, zp
-          )
-        output_tensors[action.target_name] = (target_var, pack)
-        output_tensors[action.target_name + '_quantized_scale'] = (scale, False)
+      if backend == 'cpu' and pack:
+        target_var, scale, zp = quantization_util.update_to_uint4(
+            target_var, scale, zp
+        )
+      output_tensors[action.target_name] = (target_var, pack)
+      output_tensors[action.target_name + '_quantized_scale'] = (scale, False)
+      if zp is not None:
         output_tensors[action.target_name + '_quantized_zp'] = (zp, False)
     else:
       output_tensors[action.target_name] = (action.tensor_value, False)
@@ -170,7 +172,7 @@ def convert_checkpoint(config: ConversionConfig) -> None:
   """Converts the checkpoint to tflite file."""
   logging.info('input folder: %s', config.input_ckpt)
 
-  if config.model_type == 'GEMMA_2B':
+  if config.model_type == 'GEMMA_2B' or config.model_type == 'GEMINI_XXS':
     vocab_model_path = config.vocab_model_file
   else:
     vocab_model_path = convert_bpe_vocab(

mediapipe/tasks/python/genai/converter/weight_bins_writer.py

@@ -18,6 +18,7 @@ import contextlib
 import os
 from typing import Dict, Tuple
 
+from jax import numpy as jnp
 import numpy as np
 
 from mediapipe.tasks.python.genai.converter import converter_base
@@ -65,7 +66,7 @@ class WeightBinsWriter(converter_base.ModelWriterBase):
         output = np.expand_dims(np.ravel(output), axis=-1)
         # Extra pack needed for 4 bit. We always pack the weights along the
         # first dimension since the tensor has already been squeezed.
-        output = quantization_util.pack_4bit(output, 0)
+        output = quantization_util.pack_4bit(output, 0, jnp.int8)
       if 'combined_qkv' in var_name:
         var_name = removeprefix(var_name, 'mld_vars.')
         var_name_q = var_name.replace('combined_qkv', 'q')

{mediapipe_nightly-0.10.10.post20240219.dist-info → mediapipe_nightly-0.10.10.post20240221.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mediapipe-nightly
-Version: 0.10.10.post20240219
+Version: 0.10.10.post20240221
 Summary: MediaPipe is the simplest way for researchers and developers to build world-class ML solutions and applications for mobile, edge, cloud and the web.
 Home-page: https://github.com/google/mediapipe
 Author: The MediaPipe Authors

{mediapipe_nightly-0.10.10.post20240219.dist-info → mediapipe_nightly-0.10.10.post20240221.dist-info}/RECORD

@@ -1,4 +1,4 @@
-mediapipe/__init__.py,sha256=2tmiU4WuedIsCWETIW7y0Ju2PlK3iUD4MD6strxoZK8,816
+mediapipe/__init__.py,sha256=HMF8PlJekLmUJA-rh8bf9X9ODy9JqwX9L3rOdHZMLZg,816
 mediapipe/tasks/__init__.py,sha256=sVJS2p8J2PNVl8DLRPVY6KLpHenP_z3QVPRU0x_iL5g,571
 mediapipe/tasks/python/__init__.py,sha256=wIM_WOWboOVI1MeehN8fkN_DjoA0MEBVw5mShAd8AS4,858
 mediapipe/tasks/python/benchmark/__init__.py,sha256=epEucluzX0HinwBZoS7Tgb19j_qgfTuBf-vBkqemch8,587
@@ -90,9 +90,9 @@ mediapipe/tasks/python/genai/converter/converter_base.py,sha256=nRIhPYUg2pkUZcmw
 mediapipe/tasks/python/genai/converter/pytorch_converter.py,sha256=RiuCr2aXbZVn3Ssa3OcSx_RQK56rfZRCpvG7qYIji4w,10675
 mediapipe/tasks/python/genai/converter/__init__.py,sha256=jfUkinDJR5BVldnbJMbo5vIr2Xc5Z4TTnaCJTNoAUvg,893
 mediapipe/tasks/python/genai/converter/converter_factory.py,sha256=2K16PZBQym0WhXM2HOdBMHMugykohoD4OTaOIo-UKko,2928
-mediapipe/tasks/python/genai/converter/llm_converter.py,sha256=_S8fj-WAA6OMQFj_QZPChlfkJ4gp_DE2t8i4pwJAHjI,7782
+mediapipe/tasks/python/genai/converter/llm_converter.py,sha256=srOhUwscJ6p4kICoi7bBdvkes5RvVfuHoUzh1HRsU6k,7827
 mediapipe/tasks/python/genai/converter/quantization_util_test.py,sha256=ICujhTFeREGuHGmNk1PlBpf1AUThFvv-Wl5UuZ-xWAk,9060
-mediapipe/tasks/python/genai/converter/weight_bins_writer.py,sha256=Ar8XBs1GhVSFObqN29K754iVZ_WSxe4k0Rty1a4vla0,4308
+mediapipe/tasks/python/genai/converter/weight_bins_writer.py,sha256=qrsjTWw99u-VDNhtHbnbDfMypx4sGfGyY8mMBXYnQtA,4347
 mediapipe/tasks/python/metadata/metadata.py,sha256=EECQnM-Af0angD60jaBBOuNMgt7HExH6SqVtVMFNHGc,33763
 mediapipe/tasks/python/metadata/metadata_displayer_cli.py,sha256=tLhF0B1mXG0igFTA9nPh8t1efRpRw2hQ00XpTPYdk_o,1202
 mediapipe/tasks/python/metadata/__init__.py,sha256=YGHXQMz1ZGPcNgSXggu03b0USZKE8d9Xqvn6NDUl898,586
@@ -301,7 +301,7 @@ mediapipe/util/analytics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZ
 mediapipe/util/analytics/mediapipe_log_extension_pb2.py,sha256=UX1x8jckc8NQbUz-SXOKkb8co3JNIRX3rfaMs4_Dz8g,4319
 mediapipe/util/analytics/mediapipe_logging_enums_pb2.py,sha256=9pxs-DNSQnXQe7E_LKokGDbD_G1FpmzzJw1jzex78lU,3781
 mediapipe/python/solution_base.py,sha256=48PszV6dCssQUkyPBaqhqVehVjPtskawtGUc54k10DM,25624
-mediapipe/python/_framework_bindings.cpython-39-darwin.so,sha256=KsYXkcZThsv3jnnPILzHwN3qiOoX6OJqp5KjridG3O8,78070371
+mediapipe/python/_framework_bindings.cpython-39-darwin.so,sha256=SWB99-X-zSqtkG-A8bFcNd0JsyQT8MSeaFtsHkUNPfE,78070883
 mediapipe/python/timestamp_test.py,sha256=oWKTZMsV586jH57OBV30rihcymETyGC29VbYURNLJQQ,2528
 mediapipe/python/image_frame_test.py,sha256=ZSjdE-an2t8i6MiA4_Xri91VMH5_CCx45fjhWUQptMY,8602
 mediapipe/python/__init__.py,sha256=xTuq4e55ofMUGUpV81ZbVzYHmjMmzt9dBOfS4ugFN1c,1428
@@ -377,7 +377,7 @@ mediapipe/framework/stream_handler/fixed_size_input_stream_handler_pb2.py,sha256
 mediapipe/framework/stream_handler/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mediapipe/framework/stream_handler/timestamp_align_input_stream_handler_pb2.py,sha256=Te6j6EgjmL5w3qs4kk5SnkJILDkua9cNQM0GQfNv2PY,1604
 mediapipe/framework/stream_handler/sync_set_input_stream_handler_pb2.py,sha256=ld5VHlpCHO1NFaEjXMsRxAoiDKzsd1cvmQMTqTWcNwU,1811
-mediapipe/model_maker/__init__.py,sha256=iV39VJEvbJBXI4cjXxnfVrU5CTvUtNx3cPyhCoVQPk4,1212
+mediapipe/model_maker/__init__.py,sha256=IswWcmeOPaRBNNdVp3K6m_ISXrI9_Enxnp-1kiaUEKs,1153
 mediapipe/model_maker/setup.py,sha256=GaOQVu7uTMp3gjpDHIVV45GEnLvSrgvg-BJoJAJXMbI,4064
 mediapipe/examples/__init__.py,sha256=bglKd2k2C7QGT1i-vstURXPJX2Cvq9FO9opr6cVeBp0,571
 mediapipe/examples/desktop/__init__.py,sha256=bglKd2k2C7QGT1i-vstURXPJX2Cvq9FO9opr6cVeBp0,571
@@ -559,8 +559,8 @@ mediapipe/modules/selfie_segmentation/selfie_segmentation.tflite,sha256=nuFo7HyP
 mediapipe/modules/palm_detection/palm_detection_full.tflite,sha256=GxTpQixq0AbN5lgaRsi5DdVzwHq385NLVYnnzqP4mlQ,2339846
 mediapipe/modules/palm_detection/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mediapipe/modules/palm_detection/palm_detection_lite.tflite,sha256=6aSq3fkN2laocjUwPPAOTC0_socl9o_Yh3KZfayQXBg,1985440
-mediapipe_nightly-0.10.10.post20240219.dist-info/RECORD,,
-mediapipe_nightly-0.10.10.post20240219.dist-info/LICENSE,sha256=hwfu8FM5h-_FsVXWR2HutuIHk_ULm9Gmja0c9HGdDtg,12331
-mediapipe_nightly-0.10.10.post20240219.dist-info/WHEEL,sha256=Roq3jd81zxBo_pMxKZzXrRts-yky38mkvNnjvr99fPU,109
-mediapipe_nightly-0.10.10.post20240219.dist-info/top_level.txt,sha256=LG-epD1oIiiHFRqLp--7jacjB3dbx2RfMcLYjCIhmxU,175
-mediapipe_nightly-0.10.10.post20240219.dist-info/METADATA,sha256=BBLlLK_MBEDboR4WEhiNqEiQWw7KF1RA6ohuuTA_3Kk,9672
+mediapipe_nightly-0.10.10.post20240221.dist-info/RECORD,,
+mediapipe_nightly-0.10.10.post20240221.dist-info/LICENSE,sha256=hwfu8FM5h-_FsVXWR2HutuIHk_ULm9Gmja0c9HGdDtg,12331
+mediapipe_nightly-0.10.10.post20240221.dist-info/WHEEL,sha256=Roq3jd81zxBo_pMxKZzXrRts-yky38mkvNnjvr99fPU,109
+mediapipe_nightly-0.10.10.post20240221.dist-info/top_level.txt,sha256=LG-epD1oIiiHFRqLp--7jacjB3dbx2RfMcLYjCIhmxU,175
+mediapipe_nightly-0.10.10.post20240221.dist-info/METADATA,sha256=n5NsAbtFC7eTYg8PVJytALqX5qMVya42iexL_nytaC4,9672