mediapipe-nightly 0.10.21.post20250114__cp312-cp312-manylinux_2_28_x86_64.whl

mediapipe/tasks/cc/components/processors/proto/image_preprocessing_graph_options_pb2.py

@@ -0,0 +1,34 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/tasks/cc/components/processors/proto/image_preprocessing_graph_options.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from mediapipe.calculators.tensor import image_to_tensor_calculator_pb2 as mediapipe_dot_calculators_dot_tensor_dot_image__to__tensor__calculator__pb2
+from mediapipe.framework import calculator_pb2 as mediapipe_dot_framework_dot_calculator__pb2
+try:
+  mediapipe_dot_framework_dot_calculator__options__pb2 = mediapipe_dot_framework_dot_calculator__pb2.mediapipe_dot_framework_dot_calculator__options__pb2
+except AttributeError:
+  mediapipe_dot_framework_dot_calculator__options__pb2 = mediapipe_dot_framework_dot_calculator__pb2.mediapipe.framework.calculator_options_pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nVmediapipe/tasks/cc/components/processors/proto/image_preprocessing_graph_options.proto\x12+mediapipe.tasks.components.processors.proto\x1a=mediapipe/calculators/tensor/image_to_tensor_calculator.proto\x1a$mediapipe/framework/calculator.proto\"\x91\x03\n\x1eImagePreprocessingGraphOptions\x12J\n\x17image_to_tensor_options\x18\x01 \x01(\x0b\x32).mediapipe.ImageToTensorCalculatorOptions\x12m\n\x07\x62\x61\x63kend\x18\x02 \x01(\x0e\x32S.mediapipe.tasks.components.processors.proto.ImagePreprocessingGraphOptions.Backend:\x07\x44\x45\x46\x41ULT\"8\n\x07\x42\x61\x63kend\x12\x0b\n\x07\x44\x45\x46\x41ULT\x10\x00\x12\x0f\n\x0b\x43PU_BACKEND\x10\x01\x12\x0f\n\x0bGPU_BACKEND\x10\x02\x32z\n\x03\x65xt\x12\x1c.mediapipe.CalculatorOptions\x18\x84\xf2\xed\xd9\x01 \x01(\x0b\x32K.mediapipe.tasks.components.processors.proto.ImagePreprocessingGraphOptions')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.tasks.cc.components.processors.proto.image_preprocessing_graph_options_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  DESCRIPTOR._options = None
+  _globals['_IMAGEPREPROCESSINGGRAPHOPTIONS']._serialized_start=237
+  _globals['_IMAGEPREPROCESSINGGRAPHOPTIONS']._serialized_end=638
+  _globals['_IMAGEPREPROCESSINGGRAPHOPTIONS_BACKEND']._serialized_start=458
+  _globals['_IMAGEPREPROCESSINGGRAPHOPTIONS_BACKEND']._serialized_end=514
+# @@protoc_insertion_point(module_scope)

mediapipe/tasks/cc/components/processors/proto/text_model_type_pb2.py

@@ -0,0 +1,28 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/tasks/cc/components/processors/proto/text_model_type.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nDmediapipe/tasks/cc/components/processors/proto/text_model_type.proto\x12+mediapipe.tasks.components.processors.proto\"u\n\rTextModelType\"d\n\tModelType\x12\x15\n\x11UNSPECIFIED_MODEL\x10\x00\x12\x0e\n\nBERT_MODEL\x10\x01\x12\x0f\n\x0bREGEX_MODEL\x10\x02\x12\x10\n\x0cSTRING_MODEL\x10\x03\x12\r\n\tUSE_MODEL\x10\x04')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.tasks.cc.components.processors.proto.text_model_type_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  DESCRIPTOR._options = None
+  _globals['_TEXTMODELTYPE']._serialized_start=117
+  _globals['_TEXTMODELTYPE']._serialized_end=234
+  _globals['_TEXTMODELTYPE_MODELTYPE']._serialized_start=134
+  _globals['_TEXTMODELTYPE_MODELTYPE']._serialized_end=234
+# @@protoc_insertion_point(module_scope)

mediapipe/tasks/cc/components/processors/proto/text_preprocessing_graph_options_pb2.py

@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/tasks/cc/components/processors/proto/text_preprocessing_graph_options.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from mediapipe.framework import calculator_pb2 as mediapipe_dot_framework_dot_calculator__pb2
+try:
+  mediapipe_dot_framework_dot_calculator__options__pb2 = mediapipe_dot_framework_dot_calculator__pb2.mediapipe_dot_framework_dot_calculator__options__pb2
+except AttributeError:
+  mediapipe_dot_framework_dot_calculator__options__pb2 = mediapipe_dot_framework_dot_calculator__pb2.mediapipe.framework.calculator_options_pb2
+from mediapipe.tasks.cc.components.processors.proto import text_model_type_pb2 as mediapipe_dot_tasks_dot_cc_dot_components_dot_processors_dot_proto_dot_text__model__type__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nUmediapipe/tasks/cc/components/processors/proto/text_preprocessing_graph_options.proto\x12+mediapipe.tasks.components.processors.proto\x1a$mediapipe/framework/calculator.proto\x1a\x44mediapipe/tasks/cc/components/processors/proto/text_model_type.proto\"\xac\x02\n\x1dTextPreprocessingGraphOptions\x12X\n\nmodel_type\x18\x01 \x01(\x0e\x32\x44.mediapipe.tasks.components.processors.proto.TextModelType.ModelType\x12\x13\n\x0bmax_seq_len\x18\x02 \x01(\x05\x12!\n\x19has_dynamic_input_tensors\x18\x03 \x01(\x08\x32y\n\x03\x65xt\x12\x1c.mediapipe.CalculatorOptions\x18\xbf\xbc\xb8\xe3\x01 \x01(\x0b\x32J.mediapipe.tasks.components.processors.proto.TextPreprocessingGraphOptions')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.tasks.cc.components.processors.proto.text_preprocessing_graph_options_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  DESCRIPTOR._options = None
+  _globals['_TEXTPREPROCESSINGGRAPHOPTIONS']._serialized_start=243
+  _globals['_TEXTPREPROCESSINGGRAPHOPTIONS']._serialized_end=543
+# @@protoc_insertion_point(module_scope)

mediapipe/tasks/cc/core/proto/acceleration_pb2.py

@@ -0,0 +1,28 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/tasks/cc/core/proto/acceleration.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from mediapipe.calculators.tensor import inference_calculator_pb2 as mediapipe_dot_calculators_dot_tensor_dot_inference__calculator__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n0mediapipe/tasks/cc/core/proto/acceleration.proto\x12\x1amediapipe.tasks.core.proto\x1a\x37mediapipe/calculators/tensor/inference_calculator.proto\"\xb8\x02\n\x0c\x41\x63\x63\x65leration\x12I\n\x07xnnpack\x18\x01 \x01(\x0b\x32\x36.mediapipe.InferenceCalculatorOptions.Delegate.XnnpackH\x00\x12\x41\n\x03gpu\x18\x02 \x01(\x0b\x32\x32.mediapipe.InferenceCalculatorOptions.Delegate.GpuH\x00\x12G\n\x06tflite\x18\x04 \x01(\x0b\x32\x35.mediapipe.InferenceCalculatorOptions.Delegate.TfLiteH\x00\x12\x45\n\x05nnapi\x18\x05 \x01(\x0b\x32\x34.mediapipe.InferenceCalculatorOptions.Delegate.NnapiH\x00\x42\n\n\x08\x64\x65legateB:\n%com.google.mediapipe.tasks.core.protoB\x11\x41\x63\x63\x65lerationProto')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.tasks.cc.core.proto.acceleration_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n%com.google.mediapipe.tasks.core.protoB\021AccelerationProto'
+  _globals['_ACCELERATION']._serialized_start=138
+  _globals['_ACCELERATION']._serialized_end=450
+# @@protoc_insertion_point(module_scope)

mediapipe/tasks/cc/core/proto/base_options_pb2.py

@@ -0,0 +1,30 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/tasks/cc/core/proto/base_options.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from mediapipe.gpu import gpu_origin_pb2 as mediapipe_dot_gpu_dot_gpu__origin__pb2
+from mediapipe.tasks.cc.core.proto import acceleration_pb2 as mediapipe_dot_tasks_dot_cc_dot_core_dot_proto_dot_acceleration__pb2
+from mediapipe.tasks.cc.core.proto import external_file_pb2 as mediapipe_dot_tasks_dot_cc_dot_core_dot_proto_dot_external__file__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n0mediapipe/tasks/cc/core/proto/base_options.proto\x12\x1amediapipe.tasks.core.proto\x1a\x1emediapipe/gpu/gpu_origin.proto\x1a\x30mediapipe/tasks/cc/core/proto/acceleration.proto\x1a\x31mediapipe/tasks/cc/core/proto/external_file.proto\"\xe5\x01\n\x0b\x42\x61seOptions\x12=\n\x0bmodel_asset\x18\x01 \x01(\x0b\x32(.mediapipe.tasks.core.proto.ExternalFile\x12\x1e\n\x0fuse_stream_mode\x18\x02 \x01(\x08:\x05\x66\x61lse\x12>\n\x0c\x61\x63\x63\x65leration\x18\x03 \x01(\x0b\x32(.mediapipe.tasks.core.proto.Acceleration\x12\x37\n\ngpu_origin\x18\x04 \x01(\x0e\x32\x19.mediapipe.GpuOrigin.Mode:\x08TOP_LEFTB9\n%com.google.mediapipe.tasks.core.protoB\x10\x42\x61seOptionsProto')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.tasks.cc.core.proto.base_options_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n%com.google.mediapipe.tasks.core.protoB\020BaseOptionsProto'
+  _globals['_BASEOPTIONS']._serialized_start=214
+  _globals['_BASEOPTIONS']._serialized_end=443
+# @@protoc_insertion_point(module_scope)

mediapipe/tasks/cc/core/proto/external_file_pb2.py

@@ -0,0 +1,31 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/tasks/cc/core/proto/external_file.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n1mediapipe/tasks/cc/core/proto/external_file.proto\x12\x1amediapipe.tasks.core.proto\"\xcd\x01\n\x0c\x45xternalFile\x12\x14\n\x0c\x66ile_content\x18\x01 \x01(\x0c\x12\x11\n\tfile_name\x18\x02 \x01(\t\x12L\n\x14\x66ile_descriptor_meta\x18\x03 \x01(\x0b\x32..mediapipe.tasks.core.proto.FileDescriptorMeta\x12\x46\n\x11\x66ile_pointer_meta\x18\x04 \x01(\x0b\x32+.mediapipe.tasks.core.proto.FilePointerMeta\"@\n\x12\x46ileDescriptorMeta\x12\n\n\x02\x66\x64\x18\x01 \x01(\x05\x12\x0e\n\x06length\x18\x02 \x01(\x03\x12\x0e\n\x06offset\x18\x03 \x01(\x03\"2\n\x0f\x46ilePointerMeta\x12\x0f\n\x07pointer\x18\x01 \x01(\x04\x12\x0e\n\x06length\x18\x02 \x01(\x03\x42:\n%com.google.mediapipe.tasks.core.protoB\x11\x45xternalFileProto')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.tasks.cc.core.proto.external_file_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n%com.google.mediapipe.tasks.core.protoB\021ExternalFileProto'
+  _globals['_EXTERNALFILE']._serialized_start=82
+  _globals['_EXTERNALFILE']._serialized_end=287
+  _globals['_FILEDESCRIPTORMETA']._serialized_start=289
+  _globals['_FILEDESCRIPTORMETA']._serialized_end=353
+  _globals['_FILEPOINTERMETA']._serialized_start=355
+  _globals['_FILEPOINTERMETA']._serialized_end=405
+# @@protoc_insertion_point(module_scope)

mediapipe/tasks/cc/core/proto/inference_subgraph_pb2.py

@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/tasks/cc/core/proto/inference_subgraph.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from mediapipe.framework import calculator_pb2 as mediapipe_dot_framework_dot_calculator__pb2
+try:
+  mediapipe_dot_framework_dot_calculator__options__pb2 = mediapipe_dot_framework_dot_calculator__pb2.mediapipe_dot_framework_dot_calculator__options__pb2
+except AttributeError:
+  mediapipe_dot_framework_dot_calculator__options__pb2 = mediapipe_dot_framework_dot_calculator__pb2.mediapipe.framework.calculator_options_pb2
+from mediapipe.tasks.cc.core.proto import base_options_pb2 as mediapipe_dot_tasks_dot_cc_dot_core_dot_proto_dot_base__options__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n6mediapipe/tasks/cc/core/proto/inference_subgraph.proto\x12\x1amediapipe.tasks.core.proto\x1a$mediapipe/framework/calculator.proto\x1a\x30mediapipe/tasks/cc/core/proto/base_options.proto\"\xdb\x01\n\x18InferenceSubgraphOptions\x12=\n\x0c\x62\x61se_options\x18\x01 \x01(\x0b\x32\'.mediapipe.tasks.core.proto.BaseOptions\x12\x1b\n\x13model_resources_tag\x18\x02 \x01(\t2c\n\x03\x65xt\x12\x1c.mediapipe.CalculatorOptions\x18\xbd\xe2\xd8\xd2\x01 \x01(\x0b\x32\x34.mediapipe.tasks.core.proto.InferenceSubgraphOptions')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.tasks.cc.core.proto.inference_subgraph_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  DESCRIPTOR._options = None
+  _globals['_INFERENCESUBGRAPHOPTIONS']._serialized_start=175
+  _globals['_INFERENCESUBGRAPHOPTIONS']._serialized_end=394
+# @@protoc_insertion_point(module_scope)

mediapipe/tasks/cc/core/proto/model_resources_calculator_pb2.py

@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/tasks/cc/core/proto/model_resources_calculator.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from mediapipe.framework import calculator_pb2 as mediapipe_dot_framework_dot_calculator__pb2
+try:
+  mediapipe_dot_framework_dot_calculator__options__pb2 = mediapipe_dot_framework_dot_calculator__pb2.mediapipe_dot_framework_dot_calculator__options__pb2
+except AttributeError:
+  mediapipe_dot_framework_dot_calculator__options__pb2 = mediapipe_dot_framework_dot_calculator__pb2.mediapipe.framework.calculator_options_pb2
+from mediapipe.tasks.cc.core.proto import external_file_pb2 as mediapipe_dot_tasks_dot_cc_dot_core_dot_proto_dot_external__file__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n>mediapipe/tasks/cc/core/proto/model_resources_calculator.proto\x12\x1amediapipe.tasks.core.proto\x1a$mediapipe/framework/calculator.proto\x1a\x31mediapipe/tasks/cc/core/proto/external_file.proto\"\xe8\x01\n\x1fModelResourcesCalculatorOptions\x12\x1b\n\x13model_resources_tag\x18\x01 \x01(\t\x12<\n\nmodel_file\x18\x02 \x01(\x0b\x32(.mediapipe.tasks.core.proto.ExternalFile2j\n\x03\x65xt\x12\x1c.mediapipe.CalculatorOptions\x18\xb0\xce\xf3\xd1\x01 \x01(\x0b\x32;.mediapipe.tasks.core.proto.ModelResourcesCalculatorOptions')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.tasks.cc.core.proto.model_resources_calculator_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  DESCRIPTOR._options = None
+  _globals['_MODELRESOURCESCALCULATOROPTIONS']._serialized_start=184
+  _globals['_MODELRESOURCESCALCULATOROPTIONS']._serialized_end=416
+# @@protoc_insertion_point(module_scope)

mediapipe/tasks/cc/genai/inference/calculators/detokenizer_calculator_pb2.py

@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/tasks/cc/genai/inference/calculators/detokenizer_calculator.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nKmediapipe/tasks/cc/genai/inference/calculators/detokenizer_calculator.proto\x12\x10odml.infra.proto\"\x8b\x01\n\x1c\x44\x65tokenizerCalculatorOptions\x12\x16\n\x0espm_model_file\x18\x01 \x01(\t\x12\x13\n\x0bstop_tokens\x18\x04 \x03(\t\x12\x18\n\x10num_output_heads\x18\x05 \x01(\x05\x12\x18\n\x10\x66\x61ke_weight_mode\x18\x06 \x01(\x08J\x04\x08\x02\x10\x03J\x04\x08\x03\x10\x04\x42@\n\x1b\x63om.google.odml.infra.protoB!DetokenizerCalculatorOptionsProtob\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.tasks.cc.genai.inference.calculators.detokenizer_calculator_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\033com.google.odml.infra.protoB!DetokenizerCalculatorOptionsProto'
+  _globals['_DETOKENIZERCALCULATOROPTIONS']._serialized_start=98
+  _globals['_DETOKENIZERCALCULATOROPTIONS']._serialized_end=237
+# @@protoc_insertion_point(module_scope)

mediapipe/tasks/cc/genai/inference/calculators/llm_gpu_calculator_pb2.py

@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/tasks/cc/genai/inference/calculators/llm_gpu_calculator.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from mediapipe.tasks.cc.genai.inference.proto import llm_file_metadata_pb2 as mediapipe_dot_tasks_dot_cc_dot_genai_dot_inference_dot_proto_dot_llm__file__metadata__pb2
+from mediapipe.tasks.cc.genai.inference.proto import llm_params_pb2 as mediapipe_dot_tasks_dot_cc_dot_genai_dot_inference_dot_proto_dot_llm__params__pb2
+from mediapipe.tasks.cc.genai.inference.proto import sampler_params_pb2 as mediapipe_dot_tasks_dot_cc_dot_genai_dot_inference_dot_proto_dot_sampler__params__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nGmediapipe/tasks/cc/genai/inference/calculators/llm_gpu_calculator.proto\x12\x10odml.infra.proto\x1a@mediapipe/tasks/cc/genai/inference/proto/llm_file_metadata.proto\x1a\x39mediapipe/tasks/cc/genai/inference/proto/llm_params.proto\x1a=mediapipe/tasks/cc/genai/inference/proto/sampler_params.proto\"\xaf\x05\n\x17LlmGpuCalculatorOptions\x12\x13\n\x0bweight_path\x18\x01 \x01(\t\x12N\n\x0egpu_model_info\x18\n \x01(\x0b\x32\x36.odml.infra.proto.LlmGpuCalculatorOptions.GpuModelInfo\x12\x19\n\x11num_decode_tokens\x18\x0c \x01(\x05\x12\x1b\n\x13sequence_batch_size\x18\x0e \x01(\x05\x12\x37\n\x0ellm_parameters\x18\x14 \x01(\x0b\x32\x1f.odml.infra.proto.LlmParameters\x12\x18\n\x10num_output_heads\x18\x16 \x01(\x05\x12\x12\n\nlora_ranks\x18\x1d \x03(\x05\x12;\n\x0esampler_params\x18\x1f \x01(\x0b\x32#.odml.infra.proto.SamplerParameters\x12\x1f\n\x17\x61udio_streaming_enabled\x18  \x01(\x08\x1a\xdd\x01\n\x0cGpuModelInfo\x12\x1c\n\x14\x61llow_precision_loss\x18\x01 \x01(\x08\x12\x1a\n\x12\x65nable_fast_tuning\x18\x02 \x01(\x08\x12\x1b\n\x13\x65nable_winograd_opt\x18\x03 \x01(\x08\x12\x15\n\ruse_low_power\x18\x04 \x01(\x08\x12\x1e\n\x16prefer_texture_weights\x18\x05 \x01(\x08\x12\"\n\x1a\x65nable_host_mapped_pointer\x18\x06 \x01(\x08\x12\x1b\n\x13\x64isallow_8bit_convs\x18\x07 \x01(\x08J\x04\x08\x02\x10\x03J\x04\x08\x03\x10\x04J\x04\x08\x04\x10\x05J\x04\x08\x05\x10\x06J\x04\x08\x06\x10\x07J\x04\x08\x07\x10\x08J\x04\x08\x08\x10\tJ\x04\x08\t\x10\nJ\x04\x08\x0b\x10\x0cJ\x04\x08\r\x10\x0eJ\x04\x08\x13\x10\x14J\x04\x08\x15\x10\x16J\x04\x08\x17\x10\x1cJ\x04\x08\x1e\x10\x1f\x42;\n\x1b\x63om.google.odml.infra.protoB\x1cLlmGpuCalculatorOptionsProtob\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.tasks.cc.genai.inference.calculators.llm_gpu_calculator_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\033com.google.odml.infra.protoB\034LlmGpuCalculatorOptionsProto'
+  _globals['_LLMGPUCALCULATOROPTIONS']._serialized_start=282
+  _globals['_LLMGPUCALCULATOROPTIONS']._serialized_end=969
+  _globals['_LLMGPUCALCULATOROPTIONS_GPUMODELINFO']._serialized_start=664
+  _globals['_LLMGPUCALCULATOROPTIONS_GPUMODELINFO']._serialized_end=885
+# @@protoc_insertion_point(module_scope)

mediapipe/tasks/cc/genai/inference/calculators/model_data_calculator_pb2.py

@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/tasks/cc/genai/inference/calculators/model_data_calculator.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nJmediapipe/tasks/cc/genai/inference/calculators/model_data_calculator.proto\x12\x10odml.infra.proto\"O\n\x1aModelDataCalculatorOptions\x12\x13\n\x0bweight_path\x18\x01 \x01(\t\x12\x16\n\x0espm_model_file\x18\x03 \x01(\tJ\x04\x08\x02\x10\x03\x42>\n\x1b\x63om.google.odml.infra.protoB\x1fModelDataCalculatorOptionsProtob\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.tasks.cc.genai.inference.calculators.model_data_calculator_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\033com.google.odml.infra.protoB\037ModelDataCalculatorOptionsProto'
+  _globals['_MODELDATACALCULATOROPTIONS']._serialized_start=96
+  _globals['_MODELDATACALCULATOROPTIONS']._serialized_end=175
+# @@protoc_insertion_point(module_scope)

mediapipe/tasks/cc/genai/inference/calculators/tokenizer_calculator_pb2.py

@@ -0,0 +1,29 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/tasks/cc/genai/inference/calculators/tokenizer_calculator.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nImediapipe/tasks/cc/genai/inference/calculators/tokenizer_calculator.proto\x12\x10odml.infra.proto\"\x8d\x02\n\x1aTokenizerCalculatorOptions\x12\x12\n\nmax_tokens\x18\x01 \x01(\x05\x12\x18\n\x0espm_model_file\x18\x02 \x01(\tH\x00\x12Y\n\x11tflite_model_file\x18\x04 \x01(\x0b\x32<.odml.infra.proto.TokenizerCalculatorOptions.TfLiteModelFileH\x00\x12\x16\n\x0estart_token_id\x18\x03 \x01(\x05\x1a:\n\x0fTfLiteModelFile\x12!\n\x19spm_model_key_in_metadata\x18\x02 \x01(\tJ\x04\x08\x01\x10\x02\x42\x0c\n\nmodel_fileJ\x04\x08\x05\x10\x06\x42>\n\x1b\x63om.google.odml.infra.protoB\x1fTokenizerCalculatorOptionsProtob\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.tasks.cc.genai.inference.calculators.tokenizer_calculator_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\033com.google.odml.infra.protoB\037TokenizerCalculatorOptionsProto'
+  _globals['_TOKENIZERCALCULATOROPTIONS']._serialized_start=96
+  _globals['_TOKENIZERCALCULATOROPTIONS']._serialized_end=365
+  _globals['_TOKENIZERCALCULATOROPTIONS_TFLITEMODELFILE']._serialized_start=287
+  _globals['_TOKENIZERCALCULATOROPTIONS_TFLITEMODELFILE']._serialized_end=345
+# @@protoc_insertion_point(module_scope)

mediapipe/tasks/cc/genai/inference/proto/llm_file_metadata_pb2.py

@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/tasks/cc/genai/inference/proto/llm_file_metadata.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from mediapipe.tasks.cc.genai.inference.proto import llm_params_pb2 as mediapipe_dot_tasks_dot_cc_dot_genai_dot_inference_dot_proto_dot_llm__params__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n@mediapipe/tasks/cc/genai/inference/proto/llm_file_metadata.proto\x12\x10odml.infra.proto\x1a\x39mediapipe/tasks/cc/genai/inference/proto/llm_params.proto\"\xe8\x02\n\x0fLlmFileMetadata\x12=\n\x07tensors\x18\x01 \x03(\x0b\x32,.odml.infra.proto.LlmFileMetadata.TensorInfo\x12\x35\n\x0cmodel_params\x18\x02 \x01(\x0b\x32\x1f.odml.infra.proto.LlmParameters\x12\x11\n\tlora_rank\x18\x03 \x01(\x05\x1a\xcb\x01\n\nTensorInfo\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0e\n\x06offset\x18\x02 \x01(\x04\x12\x0c\n\x04size\x18\x03 \x01(\x04\x12H\n\tdata_type\x18\x04 \x01(\x0e\x32\x35.odml.infra.proto.LlmFileMetadata.TensorInfo.DataType\"G\n\x08\x44\x61taType\x12\x0f\n\x0bUNSPECIFIED\x10\x00\x12\x0b\n\x07\x46LOAT32\x10\x01\x12\x08\n\x04INT8\x10\x02\x12\x08\n\x04INT4\x10\x03\x12\t\n\x05UINT4\x10\x04\x42\x33\n\x1b\x63om.google.odml.infra.protoB\x14LlmFileMetadataProtob\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.tasks.cc.genai.inference.proto.llm_file_metadata_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\033com.google.odml.infra.protoB\024LlmFileMetadataProto'
+  _globals['_LLMFILEMETADATA']._serialized_start=146
+  _globals['_LLMFILEMETADATA']._serialized_end=506
+  _globals['_LLMFILEMETADATA_TENSORINFO']._serialized_start=303
+  _globals['_LLMFILEMETADATA_TENSORINFO']._serialized_end=506
+  _globals['_LLMFILEMETADATA_TENSORINFO_DATATYPE']._serialized_start=435
+  _globals['_LLMFILEMETADATA_TENSORINFO_DATATYPE']._serialized_end=506
+# @@protoc_insertion_point(module_scope)

mediapipe/tasks/cc/genai/inference/proto/llm_params_pb2.py

@@ -0,0 +1,33 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/tasks/cc/genai/inference/proto/llm_params.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from mediapipe.tasks.cc.genai.inference.proto import prompt_template_pb2 as mediapipe_dot_tasks_dot_cc_dot_genai_dot_inference_dot_proto_dot_prompt__template__pb2
+from mediapipe.tasks.cc.genai.inference.proto import transformer_params_pb2 as mediapipe_dot_tasks_dot_cc_dot_genai_dot_inference_dot_proto_dot_transformer__params__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n9mediapipe/tasks/cc/genai/inference/proto/llm_params.proto\x12\x10odml.infra.proto\x1a>mediapipe/tasks/cc/genai/inference/proto/prompt_template.proto\x1a\x41mediapipe/tasks/cc/genai/inference/proto/transformer_params.proto\"\xd7\x06\n\rLlmParameters\x12G\n\x16transformer_parameters\x18\x01 \x01(\x0b\x32\'.odml.infra.proto.TransformerParameters\x12\x12\n\nvocab_size\x18\x02 \x01(\x05\x12\x18\n\x0estart_token_id\x18\x04 \x01(\x05H\x00\x12\x15\n\x0bstart_token\x18\x06 \x01(\tH\x00\x12\x13\n\x0bstop_tokens\x18\x05 \x03(\t\x12]\n\x1binput_output_normalizations\x18\x07 \x03(\x0e\x32\x38.odml.infra.proto.LlmParameters.InputOutputNormalization\x12\x39\n\x0fprompt_template\x18\x08 \x01(\x0b\x32 .odml.infra.proto.PromptTemplate\x12\x1d\n\x10num_draft_tokens\x18\t \x01(\x05H\x01\x88\x01\x01\x12\x1c\n\x0fuser_role_token\x18\x0b \x01(\tH\x02\x88\x01\x01\x12\x1e\n\x11system_role_token\x18\x0c \x01(\tH\x03\x88\x01\x01\x12\x1d\n\x10model_role_token\x18\r \x01(\tH\x04\x88\x01\x01\x12\x1b\n\x0e\x65nd_role_token\x18\x0e \x01(\tH\x05\x88\x01\x01\x12U\n\x1fsubmodel_transformer_parameters\x18\x10 \x01(\x0b\x32\'.odml.infra.proto.TransformerParametersH\x06\x88\x01\x01\"s\n\x18InputOutputNormalization\x12&\n\"INPUT_OUTPUT_NORMALIZATION_UNKNOWN\x10\x00\x12/\n+INPUT_OUTPUT_NORMALIZATION_BYTES_TO_UNICODE\x10\x01\x42\x13\n\x11start_token_unionB\x13\n\x11_num_draft_tokensB\x12\n\x10_user_role_tokenB\x14\n\x12_system_role_tokenB\x13\n\x11_model_role_tokenB\x11\n\x0f_end_role_tokenB\"\n _submodel_transformer_parametersJ\x04\x08\x03\x10\x04*\xcc\x02\n\x0cLlmModelType\x12\x1a\n\x16LLM_MODEL_TYPE_UNKNOWN\x10\x00\x12\x1f\n\x1bLLM_MODEL_TYPE_FALCON_RW_1B\x10\x05\x12\x1b\n\x17LLM_MODEL_TYPE_GEMMA_2B\x10\x06\x12\x1b\n\x17LLM_MODEL_TYPE_GEMMA_7B\x10\x0c\x12\x1c\n\x18LLM_MODEL_TYPE_GEMMA2_2B\x10\x12\x12#\n\x1fLLM_MODEL_TYPE_STABLELM_4E1T_3B\x10\x08\x12\x18\n\x14LLM_MODEL_TYPE_PHI_2\x10\x0b\x12\x1a\n\x16LLM_MODEL_TYPE_TF_LITE\x10\x64\"\x04\x08\x01\x10\x01\"\x04\x08\x02\x10\x02\"\x04\x08\x03\x10\x03\"\x04\x08\x04\x10\x04\"\x04\x08\x07\x10\x07\"\x04\x08\t\x10\t\"\x04\x08\n\x10\n\"\x04\x08\r\x10\r\"\x04\x08\x0e\x10\x0e\"\x04\x08\x0f\x10\x0f\"\x04\x08\x10\x10\x10\"\x04\x08\x11\x10\x11\"\x04\x08\x13\x10\x13\x42\x31\n\x1b\x63om.google.odml.infra.protoB\x12LLMParametersProtob\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.tasks.cc.genai.inference.proto.llm_params_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\033com.google.odml.infra.protoB\022LLMParametersProto'
+  _globals['_LLMMODELTYPE']._serialized_start=1069
+  _globals['_LLMMODELTYPE']._serialized_end=1401
+  _globals['_LLMPARAMETERS']._serialized_start=211
+  _globals['_LLMPARAMETERS']._serialized_end=1066
+  _globals['_LLMPARAMETERS_INPUTOUTPUTNORMALIZATION']._serialized_start=785
+  _globals['_LLMPARAMETERS_INPUTOUTPUTNORMALIZATION']._serialized_end=900
+# @@protoc_insertion_point(module_scope)

mediapipe/tasks/cc/genai/inference/proto/prompt_template_pb2.py

@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/tasks/cc/genai/inference/proto/prompt_template.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n>mediapipe/tasks/cc/genai/inference/proto/prompt_template.proto\x12\x10odml.infra.proto\"V\n\x0ePromptTemplate\x12\x16\n\x0esession_prefix\x18\x01 \x01(\t\x12\x15\n\rprompt_prefix\x18\x02 \x01(\t\x12\x15\n\rprompt_suffix\x18\x03 \x01(\tB2\n\x1b\x63om.google.odml.infra.protoB\x13PromptTemplateProtob\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.tasks.cc.genai.inference.proto.prompt_template_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\033com.google.odml.infra.protoB\023PromptTemplateProto'
+  _globals['_PROMPTTEMPLATE']._serialized_start=84
+  _globals['_PROMPTTEMPLATE']._serialized_end=170
+# @@protoc_insertion_point(module_scope)

mediapipe/tasks/cc/genai/inference/proto/sampler_params_pb2.py

@@ -0,0 +1,29 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/tasks/cc/genai/inference/proto/sampler_params.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n=mediapipe/tasks/cc/genai/inference/proto/sampler_params.proto\x12\x10odml.infra.proto\"\xd2\x01\n\x11SamplerParameters\x12\x36\n\x04type\x18\x01 \x01(\x0e\x32(.odml.infra.proto.SamplerParameters.Type\x12\t\n\x01k\x18\x02 \x01(\x05\x12\t\n\x01p\x18\x03 \x01(\x02\x12\x13\n\x0btemperature\x18\x04 \x01(\x02\x12\x11\n\x04seed\x18\x05 \x01(\x05H\x00\x88\x01\x01\">\n\x04Type\x12\x14\n\x10TYPE_UNSPECIFIED\x10\x00\x12\t\n\x05TOP_K\x10\x01\x12\t\n\x05TOP_P\x10\x02\x12\n\n\x06GREEDY\x10\x03\x42\x07\n\x05_seedB5\n\x1b\x63om.google.odml.infra.protoB\x16SamplerParametersProtob\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.tasks.cc.genai.inference.proto.sampler_params_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\033com.google.odml.infra.protoB\026SamplerParametersProto'
+  _globals['_SAMPLERPARAMETERS']._serialized_start=84
+  _globals['_SAMPLERPARAMETERS']._serialized_end=294
+  _globals['_SAMPLERPARAMETERS_TYPE']._serialized_start=223
+  _globals['_SAMPLERPARAMETERS_TYPE']._serialized_end=285
+# @@protoc_insertion_point(module_scope)

mediapipe/tasks/cc/genai/inference/proto/transformer_params_pb2.py

@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/tasks/cc/genai/inference/proto/transformer_params.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nAmediapipe/tasks/cc/genai/inference/proto/transformer_params.proto\x12\x10odml.infra.proto\"\xb4\x12\n\x15TransformerParameters\x12\x12\n\nbatch_size\x18\x01 \x01(\x05\x12\x16\n\x0emax_seq_length\x18\x02 \x01(\x05\x12\x15\n\rembedding_dim\x18\x03 \x01(\x05\x12\x18\n\x10hidden_dimension\x18\x04 \x01(\x05\x12\x16\n\x0ehead_dimension\x18\x05 \x01(\x05\x12\x11\n\tnum_heads\x18\x06 \x01(\x05\x12\x12\n\nnum_stacks\x18\x07 \x01(\x05\x12\x14\n\x0cnum_kv_heads\x18\t \x01(\x05\x12^\n\x17\x66\x65\x65\x64_forward_parameters\x18\x0b \x01(\x0b\x32=.odml.infra.proto.TransformerParameters.FeedForwardParameters\x12`\n\x18\x66inal_project_parameters\x18\x0c \x01(\x0b\x32>.odml.infra.proto.TransformerParameters.FinalProjectParameters\x12>\n\x08pre_norm\x18\r \x01(\x0e\x32,.odml.infra.proto.TransformerParameters.Norm\x12?\n\tpost_norm\x18\x0e \x01(\x0e\x32,.odml.infra.proto.TransformerParameters.Norm\x12@\n\nfinal_norm\x18\x0f \x01(\x0e\x32,.odml.infra.proto.TransformerParameters.Norm\x12\x62\n\x19self_attention_parameters\x18\x10 \x01(\x0b\x32?.odml.infra.proto.TransformerParameters.SelfAttentionParameters\x12+\n#skip_absolute_positional_embeddings\x18\x12 \x01(\x08\x12k\n\x1bresidual_adapter_parameters\x18\x16 \x01(\x0b\x32\x41.odml.infra.proto.TransformerParameters.ResidualAdapterParametersH\x00\x88\x01\x01\x12!\n\x14query_rescale_factor\x18\x19 \x01(\x02H\x01\x88\x01\x01\x12\x19\n\x11vision_tokens_num\x18\x1a \x01(\x05\x12\x16\n\x0emax_num_images\x18\x1c \x01(\x05\x12\x18\n\x10num_extra_stacks\x18\x1b \x01(\x05\x1a\xf1\x01\n\x15\x46\x65\x65\x64\x46orwardParameters\x12\x0f\n\x07no_bias\x18\x01 \x01(\x08\x12\x46\n\nactivation\x18\x02 \x01(\x0e\x32\x32.odml.infra.proto.TransformerParameters.Activation\x12>\n\x08pre_norm\x18\x03 \x01(\x0e\x32,.odml.infra.proto.TransformerParameters.Norm\x12?\n\tpost_norm\x18\x04 \x01(\x0e\x32,.odml.infra.proto.TransformerParameters.Norm\x1a\x41\n\x16\x46inalProjectParameters\x12\x0f\n\x07no_bias\x18\x01 \x01(\x08\x12\x16\n\x0esoft_cap_value\x18\x02 \x01(\x02\x1a\xb7\x02\n\x17SelfAttentionParameters\x12\x13\n\x0bqkv_no_bias\x18\x01 \x01(\x08\x12\x19\n\x11post_proj_no_bias\x18\x02 \x01(\x08\x12V\n\x13\x61ttention_mask_type\x18\x03 \x01(\x0e\x32\x39.odml.infra.proto.TransformerParameters.AttentionMaskType\x12\x16\n\x0esoft_cap_value\x18\x04 \x01(\x02\x12]\n\x14\x61ttention_scale_type\x18\x05 \x01(\x0e\x32:.odml.infra.proto.TransformerParameters.AttentionScaleTypeH\x00\x88\x01\x01\x42\x17\n\x15_attention_scale_typeJ\x04\x08\x06\x10\x07\x1a\x91\x01\n\x19ResidualAdapterParameters\x12V\n\x13where_to_interleave\x18\x01 \x01(\x0e\x32\x39.odml.infra.proto.TransformerParameters.WhereToInterleave\x12\x1c\n\x14\x62ottleneck_dimension\x18\x02 \x01(\x05\"O\n\x11\x41ttentionMaskType\x12\x0f\n\x0bUNSPECIFIED\x10\x00\x12\n\n\x06\x43\x41USAL\x10\x01\x12\n\n\x06PREFIX\x10\x02\x12\x11\n\rBIDIRECTIONAL\x10\x03\"S\n\nActivation\x12\x1a\n\x16\x41\x43TIVATION_UNSPECIFIED\x10\x00\x12\x08\n\x04GELU\x10\x01\x12\x08\n\x04SILU\x10\x02\x12\x08\n\x04RELU\x10\x03\x12\x0b\n\x07RELU1P5\x10\x04\"G\n\x04Norm\x12\x14\n\x10NORM_UNSPECIFIED\x10\x00\x12\x0b\n\x07NO_NORM\x10\x01\x12\x0c\n\x08RMS_NORM\x10\x02\x12\x0e\n\nLAYER_NORM\x10\x03\"\xcb\x01\n\x12\x41ttentionScaleType\x12\x1a\n\x16SCALE_TYPE_UNSPECIFIED\x10\x00\x12\x1c\n\x18SCALE_TYPE_PER_DIM_SCALE\x10\x01\x12 \n\x1cSCALE_TYPE_INV_SQRT_HEAD_DIM\x10\x02\x12-\n)SCALE_TYPE_INV_SQRT_D_MODEL_DIV_NUM_HEADS\x10\x03\x12*\n&SCALE_TYPE_RESCALE_FACTOR_INV_HEAD_DIM\x10\x04\"K\n\x11WhereToInterleave\x12\x1a\n\x16INTERLEAVE_UNSPECIFIED\x10\x00\x12\x07\n\x03\x41LL\x10\x01\x12\x11\n\rEVERY_OTHER_4\x10\x02\x42\x1e\n\x1c_residual_adapter_parametersB\x17\n\x15_query_rescale_factorJ\x04\x08\x08\x10\tJ\x04\x08\n\x10\x0bJ\x04\x08\x11\x10\x12J\x04\x08\x13\x10\x14J\x04\x08\x14\x10\x15J\x04\x08\x15\x10\x16J\x04\x08\x17\x10\x18J\x04\x08\x18\x10\x19\x42\x39\n\x1b\x63om.google.odml.infra.protoB\x1aTransformerParametersProtob\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.tasks.cc.genai.inference.proto.transformer_params_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\033com.google.odml.infra.protoB\032TransformerParametersProto'
+  _globals['_TRANSFORMERPARAMETERS']._serialized_start=88
+  _globals['_TRANSFORMERPARAMETERS']._serialized_end=2444
+  _globals['_TRANSFORMERPARAMETERS_FEEDFORWARDPARAMETERS']._serialized_start=1047
+  _globals['_TRANSFORMERPARAMETERS_FEEDFORWARDPARAMETERS']._serialized_end=1288
+  _globals['_TRANSFORMERPARAMETERS_FINALPROJECTPARAMETERS']._serialized_start=1290
+  _globals['_TRANSFORMERPARAMETERS_FINALPROJECTPARAMETERS']._serialized_end=1355
+  _globals['_TRANSFORMERPARAMETERS_SELFATTENTIONPARAMETERS']._serialized_start=1358
+  _globals['_TRANSFORMERPARAMETERS_SELFATTENTIONPARAMETERS']._serialized_end=1669
+  _globals['_TRANSFORMERPARAMETERS_RESIDUALADAPTERPARAMETERS']._serialized_start=1672
+  _globals['_TRANSFORMERPARAMETERS_RESIDUALADAPTERPARAMETERS']._serialized_end=1817
+  _globals['_TRANSFORMERPARAMETERS_ATTENTIONMASKTYPE']._serialized_start=1819
+  _globals['_TRANSFORMERPARAMETERS_ATTENTIONMASKTYPE']._serialized_end=1898
+  _globals['_TRANSFORMERPARAMETERS_ACTIVATION']._serialized_start=1900
+  _globals['_TRANSFORMERPARAMETERS_ACTIVATION']._serialized_end=1983
+  _globals['_TRANSFORMERPARAMETERS_NORM']._serialized_start=1985
+  _globals['_TRANSFORMERPARAMETERS_NORM']._serialized_end=2056
+  _globals['_TRANSFORMERPARAMETERS_ATTENTIONSCALETYPE']._serialized_start=2059
+  _globals['_TRANSFORMERPARAMETERS_ATTENTIONSCALETYPE']._serialized_end=2262
+  _globals['_TRANSFORMERPARAMETERS_WHERETOINTERLEAVE']._serialized_start=2264
+  _globals['_TRANSFORMERPARAMETERS_WHERETOINTERLEAVE']._serialized_end=2339
+# @@protoc_insertion_point(module_scope)