gimlet-api 0.0.13__py3-none-any.whl → 0.0.14__py3-none-any.whl

gml/proto/src/api/corepb/v1/model_exec_pb2.py

@@ -23,7 +23,7 @@ from gml.proto.src.common.typespb import uuid_pb2 as src_dot_common_dot_typespb_
 from gml.proto.src.api.corepb.v1 import device_info_pb2 as src_dot_api_dot_corepb_dot_v1_dot_device__info__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\"src/api/corepb/v1/model_exec.proto\x12\x18gml.internal.api.core.v1\x1a\x14gogoproto/gogo.proto\x1a$mediapipe/framework/calculator.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1egoogle/protobuf/wrappers.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a#src/api/corepb/v1/device_info.proto\"\xa7\x02\n\x0f\x42oundingBoxInfo\x12Z\n\nbox_format\x18\x01 \x01(\x0e\x32;.gml.internal.api.core.v1.BoundingBoxInfo.BoundingBoxFormatR\tboxFormat\x12%\n\x0e\x62ox_normalized\x18\x02 \x01(\x08R\rboxNormalized\"\x90\x01\n\x11\x42oundingBoxFormat\x12\x1f\n\x1b\x42OUNDING_BOX_FORMAT_UNKNOWN\x10\x00\x12\x1e\n\x1a\x42OUNDING_BOX_FORMAT_CXCYWH\x10\x01\x12\x1c\n\x18\x42OUNDING_BOX_FORMAT_YXYX\x10\x02\x12\x1c\n\x18\x42OUNDING_BOX_FORMAT_XYXY\x10\x03\"\xd8\x08\n\x16ImagePreprocessingStep\x12[\n\x04kind\x18\x01 \x01(\x0e\x32G.gml.internal.api.core.v1.ImagePreprocessingStep.ImagePreprocessingKindR\x04kind\x12s\n\x11\x63onversion_params\x18\x02 \x01(\x0b\x32\x46.gml.internal.api.core.v1.ImagePreprocessingStep.ImageConversionParamsR\x10\x63onversionParams\x12g\n\rresize_params\x18\x03 \x01(\x0b\x32\x42.gml.internal.api.core.v1.ImagePreprocessingStep.ImageResizeParamsR\x0cresizeParams\x12v\n\x12standardize_params\x18\x04 \x01(\x0b\x32G.gml.internal.api.core.v1.ImagePreprocessingStep.ImageStandardizeParamsR\x11standardizeParams\x1an\n\x15ImageConversionParams\x12\x14\n\x05scale\x18\x01 \x01(\x08R\x05scale\x12?\n\x0cscale_factor\x18\x02 \x01(\x0b\x32\x1c.google.protobuf.DoubleValueR\x0bscaleFactor\x1a\x90\x02\n\x11ImageResizeParams\x12\x66\n\x04kind\x18\x01 \x01(\x0e\x32R.gml.internal.api.core.v1.ImagePreprocessingStep.ImageResizeParams.ImageResizeKindR\x04kind\"\x92\x01\n\x0fImageResizeKind\x12\x1d\n\x19IMAGE_RESIZE_KIND_UNKNOWN\x10\x00\x12\x1d\n\x19IMAGE_RESIZE_KIND_STRETCH\x10\x01\x12\x1f\n\x1bIMAGE_RESIZE_KIND_LETTERBOX\x10\x02\x12 \n\x1cIMAGE_RESIZE_KIND_CENTERCROP\x10\x03\x1aH\n\x16ImageStandardizeParams\x12\x14\n\x05means\x18\x01 \x03(\x02R\x05means\x12\x18\n\x07stddevs\x18\x02 \x03(\x02R\x07stddevs\"\xbd\x01\n\x16ImagePreprocessingKind\x12$\n IMAGE_PREPROCESSING_KIND_UNKNOWN\x10\x00\x12.\n*IMAGE_PREPROCESSING_KIND_CONVERT_TO_TENSOR\x10\x01\x12#\n\x1fIMAGE_PREPROCESSING_KIND_RESIZE\x10\x02\x12(\n$IMAGE_PREPROCESSING_KIND_STANDARDIZE\x10\x03\"\xba\x15\n\x12\x44imensionSemantics\x12W\n\x04kind\x18\x01 \x01(\x0e\x32\x43.gml.internal.api.core.v1.DimensionSemantics.DimensionSemanticsKindR\x04kind\x12q\n\x14image_channel_params\x18\x02 \x01(\x0b\x32?.gml.internal.api.core.v1.DimensionSemantics.ImageChannelParamsR\x12imageChannelParams\x12\x86\x01\n\x1b\x64\x65tection_candidates_params\x18\x03 \x01(\x0b\x32\x46.gml.internal.api.core.v1.DimensionSemantics.DetectionCandidatesParamsR\x19\x64\x65tectionCandidatesParams\x12z\n\x17\x64\x65tection_output_params\x18\x04 \x01(\x0b\x32\x42.gml.internal.api.core.v1.DimensionSemantics.DetectionOutputParamsR\x15\x64\x65tectionOutputParams\x12}\n\x18segmentation_mask_params\x18\x05 \x01(\x0b\x32\x43.gml.internal.api.core.v1.DimensionSemantics.SegmentationMaskParamsR\x16segmentationMaskParams\x12j\n\x11regression_params\x18\x06 \x01(\x0b\x32=.gml.internal.api.core.v1.DimensionSemantics.RegressionParamsR\x10regressionParams\x1a\x99\x02\n\x12ImageChannelParams\x12j\n\x06\x66ormat\x18\x01 \x01(\x0e\x32R.gml.internal.api.core.v1.DimensionSemantics.ImageChannelParams.ImageChannelFormatR\x06\x66ormat\"\x96\x01\n\x12ImageChannelFormat\x12 \n\x1cIMAGE_CHANNEL_FORMAT_UNKNOWN\x10\x00\x12\x1c\n\x18IMAGE_CHANNEL_FORMAT_RGB\x10\x01\x12\x1c\n\x18IMAGE_CHANNEL_FORMAT_BGR\x10\x02\x12\"\n\x1eIMAGE_CHANNEL_FORMAT_GRAYSCALE\x10\x03\x1a=\n\x19\x44\x65tectionCandidatesParams\x12 \n\x0cis_nms_boxes\x18\x01 \x01(\x08R\nisNmsBoxes\x1a\x8b\x04\n\x15\x44\x65tectionOutputParams\x12\x7f\n\x14\x62ox_coordinate_range\x18\x01 \x01(\x0b\x32M.gml.internal.api.core.v1.DimensionSemantics.DetectionOutputParams.IndexRangeR\x12\x62oxCoordinateRange\x12H\n\nbox_format\x18\x02 \x01(\x0b\x32).gml.internal.api.core.v1.BoundingBoxInfoR\tboxFormat\x12\x30\n\x14\x62ox_confidence_index\x18\x03 \x01(\x05R\x12\x62oxConfidenceIndex\x12\x1f\n\x0b\x63lass_index\x18\x04 \x01(\x05R\nclassIndex\x12p\n\x0cscores_range\x18\x05 \x01(\x0b\x32M.gml.internal.api.core.v1.DimensionSemantics.DetectionOutputParams.IndexRangeR\x0bscoresRange\x12*\n\x11scores_are_logits\x18\x06 \x01(\x08R\x0fscoresAreLogits\x1a\x36\n\nIndexRange\x12\x14\n\x05start\x18\x01 \x01(\x05R\x05start\x12\x12\n\x04size\x18\x02 \x01(\x05R\x04size\x1a\xd1\x02\n\x16SegmentationMaskParams\x12l\n\x04kind\x18\x01 \x01(\x0e\x32X.gml.internal.api.core.v1.DimensionSemantics.SegmentationMaskParams.SegmentationMaskKindR\x04kind\"\xc8\x01\n\x14SegmentationMaskKind\x12\"\n\x1eSEGMENTATION_MASK_KIND_UNKNOWN\x10\x00\x12&\n\"SEGMENTATION_MASK_KIND_CLASS_LABEL\x10\x01\x12\x1f\n\x1bSEGMENTATION_MASK_KIND_BOOL\x10\x02\x12 \n\x1cSEGMENTATION_MASK_KIND_SCORE\x10\x03\x12!\n\x1dSEGMENTATION_MASK_KIND_LOGITS\x10\x04\x1a\\\n\x10RegressionParams\x12\x14\n\x05label\x18\x01 \x01(\tR\x05label\x12\x32\n\x05scale\x18\x02 \x01(\x0b\x32\x1c.google.protobuf.DoubleValueR\x05scale\"\xcc\x05\n\x16\x44imensionSemanticsKind\x12$\n DIMENSION_SEMANTICS_KIND_UNKNOWN\x10\x00\x12\"\n\x1e\x44IMENSION_SEMANTICS_KIND_BATCH\x10\x01\x12#\n\x1f\x44IMENSION_SEMANTICS_KIND_IGNORE\x10\x02\x12*\n&DIMENSION_SEMANTICS_KIND_IMAGE_CHANNEL\x10\x03\x12)\n%DIMENSION_SEMANTICS_KIND_IMAGE_HEIGHT\x10\x04\x12(\n$DIMENSION_SEMANTICS_KIND_IMAGE_WIDTH\x10\x05\x12\x31\n-DIMENSION_SEMANTICS_KIND_DETECTION_CANDIDATES\x10\x06\x12-\n)DIMENSION_SEMANTICS_KIND_DETECTION_OUTPUT\x10\x07\x12)\n%DIMENSION_SEMANTICS_KIND_CLASS_SCORES\x10\x08\x12\x36\n2DIMENSION_SEMANTICS_KIND_SEGMENTATION_MASK_CHANNEL\x10\t\x12)\n%DIMENSION_SEMANTICS_KIND_CLASS_LABELS\x10\n\x12-\n)DIMENSION_SEMANTICS_KIND_REGRESSION_VALUE\x10\x0b\x12#\n\x1f\x44IMENSION_SEMANTICS_KIND_TOKENS\x10\x0c\x12+\n\'DIMENSION_SEMANTICS_KIND_ATTENTION_MASK\x10\r\x12)\n%DIMENSION_SEMANTICS_KIND_VOCAB_LOGITS\x10\x0e\x12&\n\"DIMENSION_SEMANTICS_KIND_EMBEDDING\x10\x0f\"\xe9\x02\n\x0fTensorSemantics\x12L\n\ndimensions\x18\x01 \x03(\x0b\x32,.gml.internal.api.core.v1.DimensionSemanticsR\ndimensions\x12Q\n\x04kind\x18\x02 \x01(\x0e\x32=.gml.internal.api.core.v1.TensorSemantics.TensorSemanticsKindR\x04kind\"\xb4\x01\n\x13TensorSemanticsKind\x12!\n\x1dTENSOR_SEMANTICS_KIND_UNKNOWN\x10\x00\x12 \n\x1cTENSOR_SEMANTICS_KIND_UNUSED\x10\x01\x12#\n\x1fTENSOR_SEMANTICS_KIND_DIMENSION\x10\x02\x12\x33\n/TENSOR_SEMANTICS_KIND_ATTENTION_KEY_VALUE_CACHE\x10\x03\"6\n\x10GenerationConfig\x12\"\n\reos_token_ids\x18\x01 \x03(\x03R\x0b\x65osTokenIds\"J\n\rKVCacheConfig\x12\x39\n\x04kind\x18\x01 \x01(\x0e\x32%.gml.internal.api.core.v1.KVCacheKindR\x04kind\"i\n\x05Model\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12\x37\n\x04info\x18\x02 \x01(\x0b\x32#.gml.internal.api.core.v1.ModelInfoR\x04info\";\n\x0bPackageInfo\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x18\n\x07version\x18\x02 \x01(\tR\x07version\"[\n\x0fTracingMetadata\x12H\n\x0cpackage_info\x18\x01 \x03(\x0b\x32%.gml.internal.api.core.v1.PackageInfoR\x0bpackageInfo\"\x93\x01\n\x11\x44\x65\x66\x61ultRopeConfig\x12\x14\n\x05theta\x18\x01 \x01(\x01R\x05theta\x12\x36\n\x17max_position_embeddings\x18\x02 \x01(\x03R\x15maxPositionEmbeddings\x12\x30\n\x14rotary_embedding_dim\x18\x03 \x01(\x03R\x12rotaryEmbeddingDim\"\xc5\x02\n\x10Llama3RopeConfig\x12\x14\n\x05theta\x18\x01 \x01(\x01R\x05theta\x12\x36\n\x17max_position_embeddings\x18\x02 \x01(\x03R\x15maxPositionEmbeddings\x12\x30\n\x14rotary_embedding_dim\x18\x03 \x01(\x03R\x12rotaryEmbeddingDim\x12\x16\n\x06\x66\x61\x63tor\x18\x04 \x01(\x01R\x06\x66\x61\x63tor\x12(\n\x10high_freq_factor\x18\x05 \x01(\x01R\x0ehighFreqFactor\x12&\n\x0flow_freq_factor\x18\x06 \x01(\x01R\rlowFreqFactor\x12G\n original_max_position_embeddings\x18\x07 \x01(\x03R\x1doriginalMaxPositionEmbeddings\"\x95\x02\n\x17PositionEmbeddingConfig\x12\x43\n\x04kind\x18\x01 \x01(\x0e\x32/.gml.internal.api.core.v1.PositionEmbeddingKindR\x04kind\x12[\n\x13\x64\x65\x66\x61ult_rope_config\x18\x02 \x01(\x0b\x32+.gml.internal.api.core.v1.DefaultRopeConfigR\x11\x64\x65\x66\x61ultRopeConfig\x12X\n\x12llama3_rope_config\x18\x03 \x01(\x0b\x32*.gml.internal.api.core.v1.Llama3RopeConfigR\x10llama3RopeConfig\"\x82\x01\n\x11TransformerConfig\x12m\n\x19position_embedding_config\x18\x01 \x01(\x0b\x32\x31.gml.internal.api.core.v1.PositionEmbeddingConfigR\x17positionEmbeddingConfig\"\x80\r\n\tModelInfo\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x41\n\x04kind\x18\x02 \x01(\x0e\x32-.gml.internal.api.core.v1.ModelInfo.ModelKindR\x04kind\x12N\n\x06\x66ormat\x18\x03 \x01(\x0e\x32\x36.gml.internal.api.core.v1.ModelInfo.ModelStorageFormatR\x06\x66ormat\x12T\n\x0b\x66ile_assets\x18\x04 \x03(\x0b\x32\x33.gml.internal.api.core.v1.ModelInfo.FileAssetsEntryR\nfileAssets\x12_\n\x16input_tensor_semantics\x18\x05 \x03(\x0b\x32).gml.internal.api.core.v1.TensorSemanticsR\x14inputTensorSemantics\x12\x61\n\x17output_tensor_semantics\x18\x06 \x03(\x0b\x32).gml.internal.api.core.v1.TensorSemanticsR\x15outputTensorSemantics\x12!\n\x0c\x63lass_labels\x18\x64 \x03(\tR\x0b\x63lassLabels\x12\x46\n\tbbox_info\x18\x65 \x01(\x0b\x32).gml.internal.api.core.v1.BoundingBoxInfoR\x08\x62\x62oxInfo\x12l\n\x19image_preprocessing_steps\x18\x66 \x03(\x0b\x32\x30.gml.internal.api.core.v1.ImagePreprocessingStepR\x17imagePreprocessingSteps\x12W\n\x11generation_config\x18g \x01(\x0b\x32*.gml.internal.api.core.v1.GenerationConfigR\x10generationConfig\x12O\n\x0fkv_cache_config\x18h \x01(\x0b\x32\'.gml.internal.api.core.v1.KVCacheConfigR\rkvCacheConfig\x12T\n\x10tracing_metadata\x18i \x01(\x0b\x32).gml.internal.api.core.v1.TracingMetadataR\x0ftracingMetadata\x12Z\n\x12transformer_config\x18j \x01(\x0b\x32+.gml.internal.api.core.v1.TransformerConfigR\x11transformerConfig\x1aN\n\x0f\x46ileAssetsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDR\x05value:\x02\x38\x01\"\xd8\x01\n\tModelKind\x12\x16\n\x12MODEL_KIND_UNKNOWN\x10\x00\x12\x14\n\x10MODEL_KIND_TORCH\x10\x01\x12\x1a\n\x16MODEL_KIND_TORCHSCRIPT\x10\x02\x12\x13\n\x0fMODEL_KIND_ONNX\x10\x03\x12\x15\n\x11MODEL_KIND_TFLITE\x10\x04\x12\x17\n\x13MODEL_KIND_OPENVINO\x10\x05\x12$\n MODEL_KIND_HUGGINGFACE_TOKENIZER\x10\x06\x12\x16\n\x12MODEL_KIND_HAILORT\x10\x07\"\xd1\x02\n\x12ModelStorageFormat\x12 \n\x1cMODEL_STORAGE_FORMAT_UNKNOWN\x10\x00\x12&\n\"MODEL_STORAGE_FORMAT_MLIR_BYTECODE\x10\x01\x12\"\n\x1eMODEL_STORAGE_FORMAT_MLIR_TEXT\x10\x02\x12!\n\x1dMODEL_STORAGE_FORMAT_PROTOBUF\x10\x03\x12#\n\x1fMODEL_STORAGE_FORMAT_PROTO_TEXT\x10\x04\x12#\n\x1fMODEL_STORAGE_FORMAT_FLATBUFFER\x10\x05\x12!\n\x1dMODEL_STORAGE_FORMAT_OPENVINO\x10\x06\x12\x1f\n\x1bMODEL_STORAGE_FORMAT_OPAQUE\x10\x07\x12\x1c\n\x18MODEL_STORAGE_FORMAT_HEF\x10\x08\"\xa8\x02\n\x04Node\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x45\n\x04kind\x18\x02 \x01(\x0e\x32\x31.gml.internal.api.core.v1.LogicalPipelineNodeKindR\x04kind\x12H\n\nattributes\x18\x03 \x03(\x0b\x32(.gml.internal.api.core.v1.NodeAttributesR\nattributes\x12;\n\x06inputs\x18\x04 \x03(\x0b\x32#.gml.internal.api.core.v1.NodeInputR\x06inputs\x12>\n\x07outputs\x18\x05 \x03(\x0b\x32$.gml.internal.api.core.v1.NodeOutputR\x07outputs\"\x1e\n\x08ParamRef\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\"\xab\x04\n\x05Value\x12!\n\x0bstring_data\x18\x01 \x01(\tH\x00R\nstringData\x12\x1f\n\nint64_data\x18\x02 \x01(\x03H\x00R\tint64Data\x12!\n\x0b\x64ouble_data\x18\x03 \x01(\x01H\x00R\ndoubleData\x12\x1d\n\tbool_data\x18\x04 \x01(\x08H\x00R\x08\x62oolData\x12I\n\x0blambda_data\x18\x05 \x01(\x0b\x32&.gml.internal.api.core.v1.Value.LambdaH\x00R\nlambdaData\x12I\n\nmodel_data\x18\x06 \x01(\x0b\x32(.gml.internal.api.core.v1.Value.ModelRefH\x00R\tmodelData\x12\x43\n\nparam_data\x18\x07 \x01(\x0b\x32\".gml.internal.api.core.v1.ParamRefH\x00R\tparamData\x1aG\n\x08ModelRef\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\'\n\x02id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x1ap\n\x06Lambda\x12\x16\n\x06inputs\x18\x01 \x03(\tR\x06inputs\x12\x18\n\x07outputs\x18\x02 \x03(\tR\x07outputs\x12\x34\n\x05nodes\x18\x03 \x03(\x0b\x32\x1e.gml.internal.api.core.v1.NodeR\x05nodesB\x06\n\x04\x64\x61ta\"[\n\x0eNodeAttributes\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x35\n\x05value\x18\x02 \x01(\x0b\x32\x1f.gml.internal.api.core.v1.ValueR\x05value\"\x9c\x03\n\tNodeInput\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x45\n\x0bparam_value\x18\x02 \x01(\x0b\x32\".gml.internal.api.core.v1.ParamRefH\x00R\nparamValue\x12_\n\x11node_output_value\x18\x03 \x01(\x0b\x32\x31.gml.internal.api.core.v1.NodeInput.NodeOutputRefH\x00R\x0fnodeOutputValue\x12\x62\n\x12lambda_input_value\x18\x04 \x01(\x0b\x32\x32.gml.internal.api.core.v1.NodeInput.LambdaInputRefH\x00R\x10lambdaInputValue\x1a@\n\rNodeOutputRef\x12\x1b\n\tnode_name\x18\x01 \x01(\tR\x08nodeName\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x1a$\n\x0eLambdaInputRef\x12\x12\n\x04name\x18\x01 \x01(\tR\x04nameB\x07\n\x05value\" \n\nNodeOutput\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\"a\n\x05Param\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x44\n\rdefault_value\x18\x02 \x01(\x0b\x32\x1f.gml.internal.api.core.v1.ValueR\x0c\x64\x65\x66\x61ultValue\"\xbc\x01\n\x0fLogicalPipeline\x12\x37\n\x06params\x18\x01 \x03(\x0b\x32\x1f.gml.internal.api.core.v1.ParamR\x06params\x12\x34\n\x05nodes\x18\x02 \x03(\x0b\x32\x1e.gml.internal.api.core.v1.NodeR\x05nodes\x12:\n\tmodel_ids\x18\x03 \x03(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08ModelIDsR\x08modelIds\"\xab\x04\n\x12PipelineDeployment\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12V\n\x13logical_pipeline_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x15\xe2\xde\x1f\x11LogicalPipelineIDR\x11logicalPipelineId\x12\x37\n\x08\x66leet_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0b\xe2\xde\x1f\x07\x46leetIDR\x07\x66leetId\x12\x39\n\ncreated_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt\x12\x18\n\x07version\x18\x06 \x01(\x03R\x07version\x12\x44\n\x04spec\x18\x07 \x01(\x0b\x32\x30.gml.internal.api.core.v1.PipelineDeploymentSpecR\x04spec\x12J\n\x06status\x18\x08 \x01(\x0b\x32\x32.gml.internal.api.core.v1.PipelineDeploymentStatusR\x06status\x12\x39\n\ndeleted_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tdeletedAt\"\xb1\x01\n\x16PipelineDeploymentSpec\x12=\n\x05state\x18\x01 \x01(\x0e\x32\'.gml.internal.api.core.v1.PipelineStateR\x05state\x12X\n\x0escaling_config\x18\x02 \x01(\x0b\x32\x31.gml.internal.api.core.v1.DeploymentScalingConfigR\rscalingConfig\"\xbd\x01\n\x17\x44\x65ploymentScalingConfig\x12z\n\x12replicate_per_node\x18\x01 \x01(\x0b\x32J.gml.internal.api.core.v1.DeploymentScalingConfig.ReplicatePerNodeStrategyH\x00R\x10replicatePerNode\x1a\x1a\n\x18ReplicatePerNodeStrategyB\n\n\x08strategy\"q\n\x18PipelineDeploymentStatus\x12=\n\x05state\x18\x01 \x01(\x0e\x32\'.gml.internal.api.core.v1.PipelineStateR\x05state\x12\x16\n\x06reason\x18\x02 \x01(\tR\x06reason\"\xdc\x03\n\x10PhysicalPipeline\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12_\n\x16pipeline_deployment_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x18\xe2\xde\x1f\x14PipelineDeploymentIDR\x14pipelineDeploymentId\x12:\n\tdevice_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\x12\x39\n\ncreated_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt\x12\x42\n\x04spec\x18\x06 \x01(\x0b\x32..gml.internal.api.core.v1.PhysicalPipelineSpecR\x04spec\x12H\n\x06status\x18\x07 \x01(\x0b\x32\x30.gml.internal.api.core.v1.PhysicalPipelineStatusR\x06status\"\xcd\x02\n\x14PhysicalPipelineSpec\x12=\n\x05state\x18\x01 \x01(\x0e\x32\'.gml.internal.api.core.v1.PipelineStateR\x05state\x12\x18\n\x07version\x18\x02 \x01(\x03R\x07version\x12=\n\x05graph\x18\x03 \x01(\x0b\x32\'.gml.internal.api.core.v1.ExecutionSpecR\x05graph\x12\x30\n\x14\x64\x65vice_resource_hash\x18\x05 \x01(\tR\x12\x64\x65viceResourceHash\x12M\n\x0c\x65xec_runtime\x18\x06 \x01(\x0e\x32*.gml.internal.api.core.v1.ModelRuntimeTypeR\x0b\x65xecRuntime\x12\x1c\n\x07runtime\x18\x04 \x01(\tB\x02\x18\x01R\x07runtime\"\x88\x04\n\x16PhysicalPipelineStatus\x12=\n\x05state\x18\x01 \x01(\x0e\x32\'.gml.internal.api.core.v1.PipelineStateR\x05state\x12\x18\n\x07version\x18\x02 \x01(\x03R\x07version\x12\x16\n\x06reason\x18\x04 \x01(\tR\x06reason\x12\x30\n\x14\x64\x65vice_resource_hash\x18\x05 \x01(\tR\x12\x64\x65viceResourceHash\x12\x66\n\rruntime_specs\x18\x07 \x03(\x0b\x32\x41.gml.internal.api.core.v1.PhysicalPipelineStatus.ModelRuntimeSpecR\x0cruntimeSpecs\x12\x1c\n\x07runtime\x18\x03 \x01(\tB\x02\x18\x01R\x07runtime\x12S\n\rexec_runtimes\x18\x06 \x03(\x0e\x32*.gml.internal.api.core.v1.ModelRuntimeTypeB\x02\x18\x01R\x0c\x65xecRuntimes\x1ap\n\x10ModelRuntimeSpec\x12\x44\n\x07runtime\x18\x01 \x01(\x0e\x32*.gml.internal.api.core.v1.ModelRuntimeTypeR\x07runtime\x12\x16\n\x06\x64\x65vice\x18\x02 \x01(\tR\x06\x64\x65vice\"\x84\x01\n\x0c\x46ileResource\x12\x34\n\x07\x66ile_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\n\xe2\xde\x1f\x06\x46ileIDR\x06\x66ileId\x12\x1d\n\nsize_bytes\x18\x02 \x01(\x04R\tsizeBytes\x12\x1f\n\x0bsha256_hash\x18\x03 \x01(\tR\nsha256Hash\"\x8b\x01\n\rExecutionSpec\x12\x36\n\x05graph\x18\x01 \x01(\x0b\x32 .mediapipe.CalculatorGraphConfigR\x05graph\x12\x42\n\nmodel_spec\x18\x02 \x03(\x0b\x32#.gml.internal.api.core.v1.ModelSpecR\tmodelSpec\"\\\n\nNamedAsset\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12:\n\x04\x66ile\x18\x02 \x01(\x0b\x32&.gml.internal.api.core.v1.FileResourceR\x04\x66ile\"\xa3\x05\n\tModelSpec\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x33\n\ronnx_blob_key\x18\x02 \x01(\tB\x0f\xe2\xde\x1f\x0bONNXBlobKeyR\x0bonnxBlobKey\x12Q\n\tonnx_file\x18\x03 \x01(\x0b\x32&.gml.internal.api.core.v1.FileResourceB\x0c\xe2\xde\x1f\x08ONNXFileR\x08onnxFile\x12\x45\n\x0bnamed_asset\x18\x04 \x03(\x0b\x32$.gml.internal.api.core.v1.NamedAssetR\nnamedAsset\x12M\n\x0c\x65xec_runtime\x18\x33 \x01(\x0e\x32*.gml.internal.api.core.v1.ModelRuntimeTypeR\x0b\x65xecRuntime\x12%\n\x0e\x61\x63\x63\x65lerator_id\x18\x34 \x01(\tR\racceleratorId\x12\x62\n\rtensorrt_spec\x18\x64 \x01(\x0b\x32+.gml.internal.api.core.v1.TensorRTModelSpecB\x10\xe2\xde\x1f\x0cTensorRTSpecR\x0ctensorrtSpec\x12\x62\n\ropenvino_spec\x18\x65 \x01(\x0b\x32+.gml.internal.api.core.v1.OpenVINOModelSpecB\x10\xe2\xde\x1f\x0cOpenVINOSpecR\x0copenvinoSpec\x12W\n\tiree_spec\x18\x66 \x01(\x0b\x32\'.gml.internal.api.core.v1.IREEModelSpecB\x11\xe2\xde\x1f\rIREEModelSpecR\x08ireeSpec\x12\x1c\n\x07runtime\x18\x32 \x01(\tB\x02\x18\x01R\x07runtime\"\xfe\x01\n\x11TensorRTModelSpec\x12h\n\x14optimization_profile\x18\x01 \x03(\x0b\x32\x35.gml.internal.api.core.v1.TensorRTOptimizationProfileR\x13optimizationProfile\x12&\n\x0f\x65ngine_blob_key\x18\x02 \x01(\tR\rengineBlobKey\x12W\n\x0fmem_pool_limits\x18\x03 \x01(\x0b\x32/.gml.internal.api.core.v1.TensorRTMemPoolLimitsR\rmemPoolLimits\"\x7f\n\x1bTensorRTOptimizationProfile\x12`\n\x12tensor_shape_range\x18\x01 \x03(\x0b\x32\x32.gml.internal.api.core.v1.TensorRTTensorShapeRangeR\x10tensorShapeRange\"w\n\x18TensorRTTensorShapeRange\x12\x1f\n\x0btensor_name\x18\x01 \x01(\tR\ntensorName\x12\x10\n\x03min\x18\x03 \x03(\x05R\x03min\x12\x10\n\x03opt\x18\x04 \x03(\x05R\x03opt\x12\x10\n\x03max\x18\x05 \x03(\x05R\x03maxJ\x04\x08\x02\x10\x03\"5\n\x15TensorRTMemPoolLimits\x12\x1c\n\tworkspace\x18\x01 \x01(\x03R\tworkspace\"\xab\x01\n\x11OpenVINOModelSpec\x12X\n\x0binput_shape\x18\x01 \x03(\x0b\x32\x37.gml.internal.api.core.v1.OpenVINOModelSpec.TensorShapeR\ninputShape\x12\x1b\n\tdevice_id\x18\x02 \x01(\tR\x08\x64\x65viceId\x1a\x1f\n\x0bTensorShape\x12\x10\n\x03\x64im\x18\x01 \x03(\x05R\x03\x64im\"M\n\rIREEModelSpec\x12\x1f\n\x0b\x64river_name\x18\x01 \x01(\tR\ndriverName\x12\x1b\n\tdevice_id\x18\x02 \x01(\x04R\x08\x64\x65viceId*_\n\x0bKVCacheKind\x12\x19\n\x15KV_CACHE_KIND_UNKNOWN\x10\x00\x12\x1c\n\x18KV_CACHE_KIND_CONTIGUOUS\x10\x01\x12\x17\n\x13KV_CACHE_KIND_PAGED\x10\x02*\x8f\x01\n\x15PositionEmbeddingKind\x12#\n\x1fPOSITION_EMBEDDING_KIND_UNKNOWN\x10\x00\x12(\n$POSITION_EMBEDDING_KIND_ROPE_DEFAULT\x10\x01\x12\'\n#POSITION_EMBEDDING_KIND_ROPE_LLAMA3\x10\x02*\x96\n\n\x17LogicalPipelineNodeKind\x12&\n\"LOGICAL_PIPELINE_NODE_KIND_UNKNOWN\x10\x00\x12,\n(LOGICAL_PIPELINE_NODE_KIND_CAMERA_SOURCE\x10\n\x12\x31\n-LOGICAL_PIPELINE_NODE_KIND_TEXT_STREAM_SOURCE\x10\x0b\x12.\n*LOGICAL_PIPELINE_NODE_KIND_ENDPOINT_SOURCE\x10\x0c\x12&\n!LOGICAL_PIPELINE_NODE_KIND_DETECT\x10\xe8\x07\x12(\n#LOGICAL_PIPELINE_NODE_KIND_CLASSIFY\x10\xe9\x07\x12\'\n\"LOGICAL_PIPELINE_NODE_KIND_SEGMENT\x10\xea\x07\x12\x33\n.LOGICAL_PIPELINE_NODE_KIND_MULTI_PURPOSE_MODEL\x10\xeb\x07\x12\'\n\"LOGICAL_PIPELINE_NODE_KIND_REGRESS\x10\xec\x07\x12%\n LOGICAL_PIPELINE_NODE_KIND_TRACK\x10\xed\x07\x12/\n*LOGICAL_PIPELINE_NODE_KIND_GENERATE_TOKENS\x10\xee\x07\x12(\n#LOGICAL_PIPELINE_NODE_KIND_TOKENIZE\x10\xef\x07\x12*\n%LOGICAL_PIPELINE_NODE_KIND_DETOKENIZE\x10\xf0\x07\x12%\n LOGICAL_PIPELINE_NODE_KIND_EMBED\x10\xf1\x07\x12\x35\n0LOGICAL_PIPELINE_NODE_KIND_TEMPLATE_CHAT_MESSAGE\x10\xf2\x07\x12.\n)LOGICAL_PIPELINE_NODE_KIND_ESTIMATE_DEPTH\x10\xf3\x07\x12,\n\'LOGICAL_PIPELINE_NODE_KIND_FOR_EACH_ROI\x10\xdf\x0b\x12-\n(LOGICAL_PIPELINE_NODE_KIND_VECTOR_SEARCH\x10\xe0\x0b\x12\x34\n/LOGICAL_PIPELINE_NODE_KIND_FILTER_MOVING_TRACKS\x10\xe1\x0b\x12\x33\n.LOGICAL_PIPELINE_NODE_KIND_EXTRACT_IMAGE_FRAME\x10\xe2\x0b\x12.\n)LOGICAL_PIPELINE_NODE_KIND_EXTRACT_PROMPT\x10\xe3\x0b\x12\x31\n,LOGICAL_PIPELINE_NODE_KIND_VIDEO_STREAM_SINK\x10\xd0\x0f\x12\x37\n2LOGICAL_PIPELINE_NODE_KIND_DETECTIONS_METRICS_SINK\x10\xd1\x0f\x12\x34\n/LOGICAL_PIPELINE_NODE_KIND_LATENCY_METRICS_SINK\x10\xd2\x0f\x12\x32\n-LOGICAL_PIPELINE_NODE_KIND_FRAME_METRICS_SINK\x10\xd3\x0f\x12\x30\n+LOGICAL_PIPELINE_NODE_KIND_TEXT_STREAM_SINK\x10\xd4\x0f\x12-\n(LOGICAL_PIPELINE_NODE_KIND_ENDPOINT_SINK\x10\xd5\x0f*\xf5\x01\n\rPipelineState\x12\x1a\n\x16PIPELINE_STATE_UNKNOWN\x10\x00\x12\x1a\n\x16PIPELINE_STATE_PENDING\x10\x01\x12\x18\n\x14PIPELINE_STATE_READY\x10\x02\x12\x1a\n\x16PIPELINE_STATE_RUNNING\x10\x03\x12\x1e\n\x1aPIPELINE_STATE_TERMINATING\x10\x04\x12\x1d\n\x19PIPELINE_STATE_TERMINATED\x10\x05\x12\x19\n\x15PIPELINE_STATE_FAILED\x10\x06\x12\x1c\n\x18PIPELINE_STATE_COMPILING\x10\x07*\xc6\x01\n\rKVCacheFormat\x12\x1b\n\x17KV_CACHE_FORMAT_UNKNOWN\x10\x00\x12:\n6KV_CACHE_FORMAT_CONTIGUOUS_SEPARATE_KV_NO_OUTPUT_SPACE\x10\x01\x12<\n8KV_CACHE_FORMAT_CONTIGUOUS_COMBINED_KV_WITH_OUTPUT_SPACE\x10\x02\x12\x1e\n\x1aKV_CACHE_FORMAT_PAGED_TLLM\x10\x03\x42/Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepbb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\"src/api/corepb/v1/model_exec.proto\x12\x18gml.internal.api.core.v1\x1a\x14gogoproto/gogo.proto\x1a$mediapipe/framework/calculator.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1egoogle/protobuf/wrappers.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a#src/api/corepb/v1/device_info.proto\"\xa7\x02\n\x0f\x42oundingBoxInfo\x12Z\n\nbox_format\x18\x01 \x01(\x0e\x32;.gml.internal.api.core.v1.BoundingBoxInfo.BoundingBoxFormatR\tboxFormat\x12%\n\x0e\x62ox_normalized\x18\x02 \x01(\x08R\rboxNormalized\"\x90\x01\n\x11\x42oundingBoxFormat\x12\x1f\n\x1b\x42OUNDING_BOX_FORMAT_UNKNOWN\x10\x00\x12\x1e\n\x1a\x42OUNDING_BOX_FORMAT_CXCYWH\x10\x01\x12\x1c\n\x18\x42OUNDING_BOX_FORMAT_YXYX\x10\x02\x12\x1c\n\x18\x42OUNDING_BOX_FORMAT_XYXY\x10\x03\"\xd8\x08\n\x16ImagePreprocessingStep\x12[\n\x04kind\x18\x01 \x01(\x0e\x32G.gml.internal.api.core.v1.ImagePreprocessingStep.ImagePreprocessingKindR\x04kind\x12s\n\x11\x63onversion_params\x18\x02 \x01(\x0b\x32\x46.gml.internal.api.core.v1.ImagePreprocessingStep.ImageConversionParamsR\x10\x63onversionParams\x12g\n\rresize_params\x18\x03 \x01(\x0b\x32\x42.gml.internal.api.core.v1.ImagePreprocessingStep.ImageResizeParamsR\x0cresizeParams\x12v\n\x12standardize_params\x18\x04 \x01(\x0b\x32G.gml.internal.api.core.v1.ImagePreprocessingStep.ImageStandardizeParamsR\x11standardizeParams\x1an\n\x15ImageConversionParams\x12\x14\n\x05scale\x18\x01 \x01(\x08R\x05scale\x12?\n\x0cscale_factor\x18\x02 \x01(\x0b\x32\x1c.google.protobuf.DoubleValueR\x0bscaleFactor\x1a\x90\x02\n\x11ImageResizeParams\x12\x66\n\x04kind\x18\x01 \x01(\x0e\x32R.gml.internal.api.core.v1.ImagePreprocessingStep.ImageResizeParams.ImageResizeKindR\x04kind\"\x92\x01\n\x0fImageResizeKind\x12\x1d\n\x19IMAGE_RESIZE_KIND_UNKNOWN\x10\x00\x12\x1d\n\x19IMAGE_RESIZE_KIND_STRETCH\x10\x01\x12\x1f\n\x1bIMAGE_RESIZE_KIND_LETTERBOX\x10\x02\x12 \n\x1cIMAGE_RESIZE_KIND_CENTERCROP\x10\x03\x1aH\n\x16ImageStandardizeParams\x12\x14\n\x05means\x18\x01 \x03(\x02R\x05means\x12\x18\n\x07stddevs\x18\x02 \x03(\x02R\x07stddevs\"\xbd\x01\n\x16ImagePreprocessingKind\x12$\n IMAGE_PREPROCESSING_KIND_UNKNOWN\x10\x00\x12.\n*IMAGE_PREPROCESSING_KIND_CONVERT_TO_TENSOR\x10\x01\x12#\n\x1fIMAGE_PREPROCESSING_KIND_RESIZE\x10\x02\x12(\n$IMAGE_PREPROCESSING_KIND_STANDARDIZE\x10\x03\"\xe5\x15\n\x12\x44imensionSemantics\x12W\n\x04kind\x18\x01 \x01(\x0e\x32\x43.gml.internal.api.core.v1.DimensionSemantics.DimensionSemanticsKindR\x04kind\x12q\n\x14image_channel_params\x18\x02 \x01(\x0b\x32?.gml.internal.api.core.v1.DimensionSemantics.ImageChannelParamsR\x12imageChannelParams\x12\x86\x01\n\x1b\x64\x65tection_candidates_params\x18\x03 \x01(\x0b\x32\x46.gml.internal.api.core.v1.DimensionSemantics.DetectionCandidatesParamsR\x19\x64\x65tectionCandidatesParams\x12z\n\x17\x64\x65tection_output_params\x18\x04 \x01(\x0b\x32\x42.gml.internal.api.core.v1.DimensionSemantics.DetectionOutputParamsR\x15\x64\x65tectionOutputParams\x12}\n\x18segmentation_mask_params\x18\x05 \x01(\x0b\x32\x43.gml.internal.api.core.v1.DimensionSemantics.SegmentationMaskParamsR\x16segmentationMaskParams\x12j\n\x11regression_params\x18\x06 \x01(\x0b\x32=.gml.internal.api.core.v1.DimensionSemantics.RegressionParamsR\x10regressionParams\x1a\x99\x02\n\x12ImageChannelParams\x12j\n\x06\x66ormat\x18\x01 \x01(\x0e\x32R.gml.internal.api.core.v1.DimensionSemantics.ImageChannelParams.ImageChannelFormatR\x06\x66ormat\"\x96\x01\n\x12ImageChannelFormat\x12 \n\x1cIMAGE_CHANNEL_FORMAT_UNKNOWN\x10\x00\x12\x1c\n\x18IMAGE_CHANNEL_FORMAT_RGB\x10\x01\x12\x1c\n\x18IMAGE_CHANNEL_FORMAT_BGR\x10\x02\x12\"\n\x1eIMAGE_CHANNEL_FORMAT_GRAYSCALE\x10\x03\x1a=\n\x19\x44\x65tectionCandidatesParams\x12 \n\x0cis_nms_boxes\x18\x01 \x01(\x08R\nisNmsBoxes\x1a\x8b\x04\n\x15\x44\x65tectionOutputParams\x12\x7f\n\x14\x62ox_coordinate_range\x18\x01 \x01(\x0b\x32M.gml.internal.api.core.v1.DimensionSemantics.DetectionOutputParams.IndexRangeR\x12\x62oxCoordinateRange\x12H\n\nbox_format\x18\x02 \x01(\x0b\x32).gml.internal.api.core.v1.BoundingBoxInfoR\tboxFormat\x12\x30\n\x14\x62ox_confidence_index\x18\x03 \x01(\x05R\x12\x62oxConfidenceIndex\x12\x1f\n\x0b\x63lass_index\x18\x04 \x01(\x05R\nclassIndex\x12p\n\x0cscores_range\x18\x05 \x01(\x0b\x32M.gml.internal.api.core.v1.DimensionSemantics.DetectionOutputParams.IndexRangeR\x0bscoresRange\x12*\n\x11scores_are_logits\x18\x06 \x01(\x08R\x0fscoresAreLogits\x1a\x36\n\nIndexRange\x12\x14\n\x05start\x18\x01 \x01(\x05R\x05start\x12\x12\n\x04size\x18\x02 \x01(\x05R\x04size\x1a\xd1\x02\n\x16SegmentationMaskParams\x12l\n\x04kind\x18\x01 \x01(\x0e\x32X.gml.internal.api.core.v1.DimensionSemantics.SegmentationMaskParams.SegmentationMaskKindR\x04kind\"\xc8\x01\n\x14SegmentationMaskKind\x12\"\n\x1eSEGMENTATION_MASK_KIND_UNKNOWN\x10\x00\x12&\n\"SEGMENTATION_MASK_KIND_CLASS_LABEL\x10\x01\x12\x1f\n\x1bSEGMENTATION_MASK_KIND_BOOL\x10\x02\x12 \n\x1cSEGMENTATION_MASK_KIND_SCORE\x10\x03\x12!\n\x1dSEGMENTATION_MASK_KIND_LOGITS\x10\x04\x1a\\\n\x10RegressionParams\x12\x14\n\x05label\x18\x01 \x01(\tR\x05label\x12\x32\n\x05scale\x18\x02 \x01(\x0b\x32\x1c.google.protobuf.DoubleValueR\x05scale\"\xf7\x05\n\x16\x44imensionSemanticsKind\x12$\n DIMENSION_SEMANTICS_KIND_UNKNOWN\x10\x00\x12\"\n\x1e\x44IMENSION_SEMANTICS_KIND_BATCH\x10\x01\x12#\n\x1f\x44IMENSION_SEMANTICS_KIND_IGNORE\x10\x02\x12*\n&DIMENSION_SEMANTICS_KIND_IMAGE_CHANNEL\x10\x03\x12)\n%DIMENSION_SEMANTICS_KIND_IMAGE_HEIGHT\x10\x04\x12(\n$DIMENSION_SEMANTICS_KIND_IMAGE_WIDTH\x10\x05\x12\x31\n-DIMENSION_SEMANTICS_KIND_DETECTION_CANDIDATES\x10\x06\x12-\n)DIMENSION_SEMANTICS_KIND_DETECTION_OUTPUT\x10\x07\x12)\n%DIMENSION_SEMANTICS_KIND_CLASS_SCORES\x10\x08\x12\x36\n2DIMENSION_SEMANTICS_KIND_SEGMENTATION_MASK_CHANNEL\x10\t\x12)\n%DIMENSION_SEMANTICS_KIND_CLASS_LABELS\x10\n\x12-\n)DIMENSION_SEMANTICS_KIND_REGRESSION_VALUE\x10\x0b\x12#\n\x1f\x44IMENSION_SEMANTICS_KIND_TOKENS\x10\x0c\x12+\n\'DIMENSION_SEMANTICS_KIND_ATTENTION_MASK\x10\r\x12)\n%DIMENSION_SEMANTICS_KIND_VOCAB_LOGITS\x10\x0e\x12&\n\"DIMENSION_SEMANTICS_KIND_EMBEDDING\x10\x0f\x12)\n%DIMENSION_SEMANTICS_KIND_POSITION_IDS\x10\x10\"\xe9\x02\n\x0fTensorSemantics\x12L\n\ndimensions\x18\x01 \x03(\x0b\x32,.gml.internal.api.core.v1.DimensionSemanticsR\ndimensions\x12Q\n\x04kind\x18\x02 \x01(\x0e\x32=.gml.internal.api.core.v1.TensorSemantics.TensorSemanticsKindR\x04kind\"\xb4\x01\n\x13TensorSemanticsKind\x12!\n\x1dTENSOR_SEMANTICS_KIND_UNKNOWN\x10\x00\x12 \n\x1cTENSOR_SEMANTICS_KIND_UNUSED\x10\x01\x12#\n\x1fTENSOR_SEMANTICS_KIND_DIMENSION\x10\x02\x12\x33\n/TENSOR_SEMANTICS_KIND_ATTENTION_KEY_VALUE_CACHE\x10\x03\"6\n\x10GenerationConfig\x12\"\n\reos_token_ids\x18\x01 \x03(\x03R\x0b\x65osTokenIds\"J\n\rKVCacheConfig\x12\x39\n\x04kind\x18\x01 \x01(\x0e\x32%.gml.internal.api.core.v1.KVCacheKindR\x04kind\"i\n\x05Model\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12\x37\n\x04info\x18\x02 \x01(\x0b\x32#.gml.internal.api.core.v1.ModelInfoR\x04info\";\n\x0bPackageInfo\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x18\n\x07version\x18\x02 \x01(\tR\x07version\"[\n\x0fTracingMetadata\x12H\n\x0cpackage_info\x18\x01 \x03(\x0b\x32%.gml.internal.api.core.v1.PackageInfoR\x0bpackageInfo\"\x93\x01\n\x11\x44\x65\x66\x61ultRopeConfig\x12\x14\n\x05theta\x18\x01 \x01(\x01R\x05theta\x12\x36\n\x17max_position_embeddings\x18\x02 \x01(\x03R\x15maxPositionEmbeddings\x12\x30\n\x14rotary_embedding_dim\x18\x03 \x01(\x03R\x12rotaryEmbeddingDim\"\xc5\x02\n\x10Llama3RopeConfig\x12\x14\n\x05theta\x18\x01 \x01(\x01R\x05theta\x12\x36\n\x17max_position_embeddings\x18\x02 \x01(\x03R\x15maxPositionEmbeddings\x12\x30\n\x14rotary_embedding_dim\x18\x03 \x01(\x03R\x12rotaryEmbeddingDim\x12\x16\n\x06\x66\x61\x63tor\x18\x04 \x01(\x01R\x06\x66\x61\x63tor\x12(\n\x10high_freq_factor\x18\x05 \x01(\x01R\x0ehighFreqFactor\x12&\n\x0flow_freq_factor\x18\x06 \x01(\x01R\rlowFreqFactor\x12G\n original_max_position_embeddings\x18\x07 \x01(\x03R\x1doriginalMaxPositionEmbeddings\"\x95\x02\n\x17PositionEmbeddingConfig\x12\x43\n\x04kind\x18\x01 \x01(\x0e\x32/.gml.internal.api.core.v1.PositionEmbeddingKindR\x04kind\x12[\n\x13\x64\x65\x66\x61ult_rope_config\x18\x02 \x01(\x0b\x32+.gml.internal.api.core.v1.DefaultRopeConfigR\x11\x64\x65\x66\x61ultRopeConfig\x12X\n\x12llama3_rope_config\x18\x03 \x01(\x0b\x32*.gml.internal.api.core.v1.Llama3RopeConfigR\x10llama3RopeConfig\"\x82\x01\n\x11TransformerConfig\x12m\n\x19position_embedding_config\x18\x01 \x01(\x0b\x32\x31.gml.internal.api.core.v1.PositionEmbeddingConfigR\x17positionEmbeddingConfig\"\xbc\r\n\tModelInfo\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x41\n\x04kind\x18\x02 \x01(\x0e\x32-.gml.internal.api.core.v1.ModelInfo.ModelKindR\x04kind\x12N\n\x06\x66ormat\x18\x03 \x01(\x0e\x32\x36.gml.internal.api.core.v1.ModelInfo.ModelStorageFormatR\x06\x66ormat\x12T\n\x0b\x66ile_assets\x18\x04 \x03(\x0b\x32\x33.gml.internal.api.core.v1.ModelInfo.FileAssetsEntryR\nfileAssets\x12_\n\x16input_tensor_semantics\x18\x05 \x03(\x0b\x32).gml.internal.api.core.v1.TensorSemanticsR\x14inputTensorSemantics\x12\x61\n\x17output_tensor_semantics\x18\x06 \x03(\x0b\x32).gml.internal.api.core.v1.TensorSemanticsR\x15outputTensorSemantics\x12!\n\x0c\x63lass_labels\x18\x64 \x03(\tR\x0b\x63lassLabels\x12\x46\n\tbbox_info\x18\x65 \x01(\x0b\x32).gml.internal.api.core.v1.BoundingBoxInfoR\x08\x62\x62oxInfo\x12l\n\x19image_preprocessing_steps\x18\x66 \x03(\x0b\x32\x30.gml.internal.api.core.v1.ImagePreprocessingStepR\x17imagePreprocessingSteps\x12W\n\x11generation_config\x18g \x01(\x0b\x32*.gml.internal.api.core.v1.GenerationConfigR\x10generationConfig\x12O\n\x0fkv_cache_config\x18h \x01(\x0b\x32\'.gml.internal.api.core.v1.KVCacheConfigR\rkvCacheConfig\x12T\n\x10tracing_metadata\x18i \x01(\x0b\x32).gml.internal.api.core.v1.TracingMetadataR\x0ftracingMetadata\x12Z\n\x12transformer_config\x18j \x01(\x0b\x32+.gml.internal.api.core.v1.TransformerConfigR\x11transformerConfig\x1aN\n\x0f\x46ileAssetsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12%\n\x05value\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDR\x05value:\x02\x38\x01\"\xef\x01\n\tModelKind\x12\x16\n\x12MODEL_KIND_UNKNOWN\x10\x00\x12\x14\n\x10MODEL_KIND_TORCH\x10\x01\x12\x1a\n\x16MODEL_KIND_TORCHSCRIPT\x10\x02\x12\x13\n\x0fMODEL_KIND_ONNX\x10\x03\x12\x15\n\x11MODEL_KIND_TFLITE\x10\x04\x12\x17\n\x13MODEL_KIND_OPENVINO\x10\x05\x12$\n MODEL_KIND_HUGGINGFACE_TOKENIZER\x10\x06\x12\x16\n\x12MODEL_KIND_HAILORT\x10\x07\x12\x15\n\x11MODEL_KIND_HABANA\x10\x08\"\xf6\x02\n\x12ModelStorageFormat\x12 \n\x1cMODEL_STORAGE_FORMAT_UNKNOWN\x10\x00\x12&\n\"MODEL_STORAGE_FORMAT_MLIR_BYTECODE\x10\x01\x12\"\n\x1eMODEL_STORAGE_FORMAT_MLIR_TEXT\x10\x02\x12!\n\x1dMODEL_STORAGE_FORMAT_PROTOBUF\x10\x03\x12#\n\x1fMODEL_STORAGE_FORMAT_PROTO_TEXT\x10\x04\x12#\n\x1fMODEL_STORAGE_FORMAT_FLATBUFFER\x10\x05\x12!\n\x1dMODEL_STORAGE_FORMAT_OPENVINO\x10\x06\x12\x1f\n\x1bMODEL_STORAGE_FORMAT_OPAQUE\x10\x07\x12\x1c\n\x18MODEL_STORAGE_FORMAT_HEF\x10\x08\x12#\n\x1fMODEL_STORAGE_FORMAT_HABANA_BIN\x10\t\"L\n\rModelInfoList\x12;\n\x06models\x18\x01 \x03(\x0b\x32#.gml.internal.api.core.v1.ModelInfoR\x06models\"\xa8\x02\n\x04Node\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x45\n\x04kind\x18\x02 \x01(\x0e\x32\x31.gml.internal.api.core.v1.LogicalPipelineNodeKindR\x04kind\x12H\n\nattributes\x18\x03 \x03(\x0b\x32(.gml.internal.api.core.v1.NodeAttributesR\nattributes\x12;\n\x06inputs\x18\x04 \x03(\x0b\x32#.gml.internal.api.core.v1.NodeInputR\x06inputs\x12>\n\x07outputs\x18\x05 \x03(\x0b\x32$.gml.internal.api.core.v1.NodeOutputR\x07outputs\"\x1e\n\x08ParamRef\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\"D\n\tValueList\x12\x37\n\x06values\x18\x01 \x03(\x0b\x32\x1f.gml.internal.api.core.v1.ValueR\x06values\"\xb0\x01\n\tValueDict\x12G\n\x06values\x18\x01 \x03(\x0b\x32/.gml.internal.api.core.v1.ValueDict.ValuesEntryR\x06values\x1aZ\n\x0bValuesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x35\n\x05value\x18\x02 \x01(\x0b\x32\x1f.gml.internal.api.core.v1.ValueR\x05value:\x02\x38\x01\"\x86\x06\n\x05Value\x12!\n\x0bstring_data\x18\x01 \x01(\tH\x00R\nstringData\x12\x1f\n\nint64_data\x18\x02 \x01(\x03H\x00R\tint64Data\x12!\n\x0b\x64ouble_data\x18\x03 \x01(\x01H\x00R\ndoubleData\x12\x1d\n\tbool_data\x18\x04 \x01(\x08H\x00R\x08\x62oolData\x12I\n\x0blambda_data\x18\x05 \x01(\x0b\x32&.gml.internal.api.core.v1.Value.LambdaH\x00R\nlambdaData\x12I\n\nmodel_data\x18\x06 \x01(\x0b\x32(.gml.internal.api.core.v1.Value.ModelRefH\x00R\tmodelData\x12\x43\n\nparam_data\x18\x07 \x01(\x0b\x32\".gml.internal.api.core.v1.ParamRefH\x00R\tparamData\x12\x44\n\nvalue_list\x18\x08 \x01(\x0b\x32#.gml.internal.api.core.v1.ValueListH\x00R\tvalueList\x12\x44\n\nvalue_dict\x18\t \x01(\x0b\x32#.gml.internal.api.core.v1.ValueDictH\x00R\tvalueDict\x12M\n\rsemantic_type\x18\n \x01(\x0e\x32&.gml.internal.api.core.v1.SemanticTypeH\x00R\x0csemanticType\x1aG\n\x08ModelRef\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\'\n\x02id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x1ap\n\x06Lambda\x12\x16\n\x06inputs\x18\x01 \x03(\tR\x06inputs\x12\x18\n\x07outputs\x18\x02 \x03(\tR\x07outputs\x12\x34\n\x05nodes\x18\x03 \x03(\x0b\x32\x1e.gml.internal.api.core.v1.NodeR\x05nodesB\x06\n\x04\x64\x61ta\"[\n\x0eNodeAttributes\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x35\n\x05value\x18\x02 \x01(\x0b\x32\x1f.gml.internal.api.core.v1.ValueR\x05value\"\x9c\x03\n\tNodeInput\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x45\n\x0bparam_value\x18\x02 \x01(\x0b\x32\".gml.internal.api.core.v1.ParamRefH\x00R\nparamValue\x12_\n\x11node_output_value\x18\x03 \x01(\x0b\x32\x31.gml.internal.api.core.v1.NodeInput.NodeOutputRefH\x00R\x0fnodeOutputValue\x12\x62\n\x12lambda_input_value\x18\x04 \x01(\x0b\x32\x32.gml.internal.api.core.v1.NodeInput.LambdaInputRefH\x00R\x10lambdaInputValue\x1a@\n\rNodeOutputRef\x12\x1b\n\tnode_name\x18\x01 \x01(\tR\x08nodeName\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x1a$\n\x0eLambdaInputRef\x12\x12\n\x04name\x18\x01 \x01(\tR\x04nameB\x07\n\x05value\" \n\nNodeOutput\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\"a\n\x05Param\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x44\n\rdefault_value\x18\x02 \x01(\x0b\x32\x1f.gml.internal.api.core.v1.ValueR\x0c\x64\x65\x66\x61ultValue\"\xbc\x01\n\x0fLogicalPipeline\x12\x37\n\x06params\x18\x01 \x03(\x0b\x32\x1f.gml.internal.api.core.v1.ParamR\x06params\x12\x34\n\x05nodes\x18\x02 \x03(\x0b\x32\x1e.gml.internal.api.core.v1.NodeR\x05nodes\x12:\n\tmodel_ids\x18\x03 \x03(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08ModelIDsR\x08modelIds\"\xab\x04\n\x12PipelineDeployment\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12V\n\x13logical_pipeline_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x15\xe2\xde\x1f\x11LogicalPipelineIDR\x11logicalPipelineId\x12\x37\n\x08\x66leet_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0b\xe2\xde\x1f\x07\x46leetIDR\x07\x66leetId\x12\x39\n\ncreated_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt\x12\x18\n\x07version\x18\x06 \x01(\x03R\x07version\x12\x44\n\x04spec\x18\x07 \x01(\x0b\x32\x30.gml.internal.api.core.v1.PipelineDeploymentSpecR\x04spec\x12J\n\x06status\x18\x08 \x01(\x0b\x32\x32.gml.internal.api.core.v1.PipelineDeploymentStatusR\x06status\x12\x39\n\ndeleted_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tdeletedAt\"\xb1\x01\n\x16PipelineDeploymentSpec\x12=\n\x05state\x18\x01 \x01(\x0e\x32\'.gml.internal.api.core.v1.PipelineStateR\x05state\x12X\n\x0escaling_config\x18\x02 \x01(\x0b\x32\x31.gml.internal.api.core.v1.DeploymentScalingConfigR\rscalingConfig\"\xbd\x01\n\x17\x44\x65ploymentScalingConfig\x12z\n\x12replicate_per_node\x18\x01 \x01(\x0b\x32J.gml.internal.api.core.v1.DeploymentScalingConfig.ReplicatePerNodeStrategyH\x00R\x10replicatePerNode\x1a\x1a\n\x18ReplicatePerNodeStrategyB\n\n\x08strategy\"q\n\x18PipelineDeploymentStatus\x12=\n\x05state\x18\x01 \x01(\x0e\x32\'.gml.internal.api.core.v1.PipelineStateR\x05state\x12\x16\n\x06reason\x18\x02 \x01(\tR\x06reason\"\xdc\x03\n\x10PhysicalPipeline\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12_\n\x16pipeline_deployment_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x18\xe2\xde\x1f\x14PipelineDeploymentIDR\x14pipelineDeploymentId\x12:\n\tdevice_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\x12\x39\n\ncreated_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt\x12\x42\n\x04spec\x18\x06 \x01(\x0b\x32..gml.internal.api.core.v1.PhysicalPipelineSpecR\x04spec\x12H\n\x06status\x18\x07 \x01(\x0b\x32\x30.gml.internal.api.core.v1.PhysicalPipelineStatusR\x06status\"\xcd\x02\n\x14PhysicalPipelineSpec\x12=\n\x05state\x18\x01 \x01(\x0e\x32\'.gml.internal.api.core.v1.PipelineStateR\x05state\x12\x18\n\x07version\x18\x02 \x01(\x03R\x07version\x12=\n\x05graph\x18\x03 \x01(\x0b\x32\'.gml.internal.api.core.v1.ExecutionSpecR\x05graph\x12\x30\n\x14\x64\x65vice_resource_hash\x18\x05 \x01(\tR\x12\x64\x65viceResourceHash\x12M\n\x0c\x65xec_runtime\x18\x06 \x01(\x0e\x32*.gml.internal.api.core.v1.ModelRuntimeTypeR\x0b\x65xecRuntime\x12\x1c\n\x07runtime\x18\x04 \x01(\tB\x02\x18\x01R\x07runtime\"\x88\x04\n\x16PhysicalPipelineStatus\x12=\n\x05state\x18\x01 \x01(\x0e\x32\'.gml.internal.api.core.v1.PipelineStateR\x05state\x12\x18\n\x07version\x18\x02 \x01(\x03R\x07version\x12\x16\n\x06reason\x18\x04 \x01(\tR\x06reason\x12\x30\n\x14\x64\x65vice_resource_hash\x18\x05 \x01(\tR\x12\x64\x65viceResourceHash\x12\x66\n\rruntime_specs\x18\x07 \x03(\x0b\x32\x41.gml.internal.api.core.v1.PhysicalPipelineStatus.ModelRuntimeSpecR\x0cruntimeSpecs\x12\x1c\n\x07runtime\x18\x03 \x01(\tB\x02\x18\x01R\x07runtime\x12S\n\rexec_runtimes\x18\x06 \x03(\x0e\x32*.gml.internal.api.core.v1.ModelRuntimeTypeB\x02\x18\x01R\x0c\x65xecRuntimes\x1ap\n\x10ModelRuntimeSpec\x12\x44\n\x07runtime\x18\x01 \x01(\x0e\x32*.gml.internal.api.core.v1.ModelRuntimeTypeR\x07runtime\x12\x16\n\x06\x64\x65vice\x18\x02 \x01(\tR\x06\x64\x65vice\"\x84\x01\n\x0c\x46ileResource\x12\x34\n\x07\x66ile_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\n\xe2\xde\x1f\x06\x46ileIDR\x06\x66ileId\x12\x1d\n\nsize_bytes\x18\x02 \x01(\x04R\tsizeBytes\x12\x1f\n\x0bsha256_hash\x18\x03 \x01(\tR\nsha256Hash\"\x8b\x01\n\rExecutionSpec\x12\x36\n\x05graph\x18\x01 \x01(\x0b\x32 .mediapipe.CalculatorGraphConfigR\x05graph\x12\x42\n\nmodel_spec\x18\x02 \x03(\x0b\x32#.gml.internal.api.core.v1.ModelSpecR\tmodelSpec\"\\\n\nNamedAsset\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12:\n\x04\x66ile\x18\x02 \x01(\x0b\x32&.gml.internal.api.core.v1.FileResourceR\x04\x66ile\"\xad\x06\n\tModelSpec\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x33\n\ronnx_blob_key\x18\x02 \x01(\tB\x0f\xe2\xde\x1f\x0bONNXBlobKeyR\x0bonnxBlobKey\x12Q\n\tonnx_file\x18\x03 \x01(\x0b\x32&.gml.internal.api.core.v1.FileResourceB\x0c\xe2\xde\x1f\x08ONNXFileR\x08onnxFile\x12\x45\n\x0bnamed_asset\x18\x04 \x03(\x0b\x32$.gml.internal.api.core.v1.NamedAssetR\nnamedAsset\x12M\n\x0c\x65xec_runtime\x18\x33 \x01(\x0e\x32*.gml.internal.api.core.v1.ModelRuntimeTypeR\x0b\x65xecRuntime\x12%\n\x0e\x61\x63\x63\x65lerator_id\x18\x34 \x01(\tR\racceleratorId\x12\'\n\x0f\x61\x63\x63\x65lerator_idx\x18\x35 \x01(\x05R\x0e\x61\x63\x63\x65leratorIdx\x12\x62\n\rtensorrt_spec\x18\x64 \x01(\x0b\x32+.gml.internal.api.core.v1.TensorRTModelSpecB\x10\xe2\xde\x1f\x0cTensorRTSpecR\x0ctensorrtSpec\x12\x62\n\ropenvino_spec\x18\x65 \x01(\x0b\x32+.gml.internal.api.core.v1.OpenVINOModelSpecB\x10\xe2\xde\x1f\x0cOpenVINOSpecR\x0copenvinoSpec\x12W\n\tiree_spec\x18\x66 \x01(\x0b\x32\'.gml.internal.api.core.v1.IREEModelSpecB\x11\xe2\xde\x1f\rIREEModelSpecR\x08ireeSpec\x12_\n\x0bhabana_spec\x18g \x01(\x0b\x32).gml.internal.api.core.v1.HabanaModelSpecB\x13\xe2\xde\x1f\x0fHabanaModelSpecR\nhabanaSpec\x12\x1c\n\x07runtime\x18\x32 \x01(\tB\x02\x18\x01R\x07runtime\"\xfe\x01\n\x11TensorRTModelSpec\x12h\n\x14optimization_profile\x18\x01 \x03(\x0b\x32\x35.gml.internal.api.core.v1.TensorRTOptimizationProfileR\x13optimizationProfile\x12&\n\x0f\x65ngine_blob_key\x18\x02 \x01(\tR\rengineBlobKey\x12W\n\x0fmem_pool_limits\x18\x03 \x01(\x0b\x32/.gml.internal.api.core.v1.TensorRTMemPoolLimitsR\rmemPoolLimits\"\x7f\n\x1bTensorRTOptimizationProfile\x12`\n\x12tensor_shape_range\x18\x01 \x03(\x0b\x32\x32.gml.internal.api.core.v1.TensorRTTensorShapeRangeR\x10tensorShapeRange\"w\n\x18TensorRTTensorShapeRange\x12\x1f\n\x0btensor_name\x18\x01 \x01(\tR\ntensorName\x12\x10\n\x03min\x18\x03 \x03(\x05R\x03min\x12\x10\n\x03opt\x18\x04 \x03(\x05R\x03opt\x12\x10\n\x03max\x18\x05 \x03(\x05R\x03maxJ\x04\x08\x02\x10\x03\"5\n\x15TensorRTMemPoolLimits\x12\x1c\n\tworkspace\x18\x01 \x01(\x03R\tworkspace\"\xab\x01\n\x11OpenVINOModelSpec\x12X\n\x0binput_shape\x18\x01 \x03(\x0b\x32\x37.gml.internal.api.core.v1.OpenVINOModelSpec.TensorShapeR\ninputShape\x12\x1b\n\tdevice_id\x18\x02 \x01(\tR\x08\x64\x65viceId\x1a\x1f\n\x0bTensorShape\x12\x10\n\x03\x64im\x18\x01 \x03(\x05R\x03\x64im\"M\n\rIREEModelSpec\x12\x1f\n\x0b\x64river_name\x18\x01 \x01(\tR\ndriverName\x12\x1b\n\tdevice_id\x18\x02 \x01(\x04R\x08\x64\x65viceId\".\n\x0fHabanaModelSpec\x12\x1b\n\tdevice_id\x18\x01 \x01(\tR\x08\x64\x65viceId*_\n\x0bKVCacheKind\x12\x19\n\x15KV_CACHE_KIND_UNKNOWN\x10\x00\x12\x1c\n\x18KV_CACHE_KIND_CONTIGUOUS\x10\x01\x12\x17\n\x13KV_CACHE_KIND_PAGED\x10\x02*\x8f\x01\n\x15PositionEmbeddingKind\x12#\n\x1fPOSITION_EMBEDDING_KIND_UNKNOWN\x10\x00\x12(\n$POSITION_EMBEDDING_KIND_ROPE_DEFAULT\x10\x01\x12\'\n#POSITION_EMBEDDING_KIND_ROPE_LLAMA3\x10\x02*\x96\n\n\x17LogicalPipelineNodeKind\x12&\n\"LOGICAL_PIPELINE_NODE_KIND_UNKNOWN\x10\x00\x12,\n(LOGICAL_PIPELINE_NODE_KIND_CAMERA_SOURCE\x10\n\x12\x31\n-LOGICAL_PIPELINE_NODE_KIND_TEXT_STREAM_SOURCE\x10\x0b\x12.\n*LOGICAL_PIPELINE_NODE_KIND_ENDPOINT_SOURCE\x10\x0c\x12&\n!LOGICAL_PIPELINE_NODE_KIND_DETECT\x10\xe8\x07\x12(\n#LOGICAL_PIPELINE_NODE_KIND_CLASSIFY\x10\xe9\x07\x12\'\n\"LOGICAL_PIPELINE_NODE_KIND_SEGMENT\x10\xea\x07\x12\x33\n.LOGICAL_PIPELINE_NODE_KIND_MULTI_PURPOSE_MODEL\x10\xeb\x07\x12\'\n\"LOGICAL_PIPELINE_NODE_KIND_REGRESS\x10\xec\x07\x12%\n LOGICAL_PIPELINE_NODE_KIND_TRACK\x10\xed\x07\x12/\n*LOGICAL_PIPELINE_NODE_KIND_GENERATE_TOKENS\x10\xee\x07\x12(\n#LOGICAL_PIPELINE_NODE_KIND_TOKENIZE\x10\xef\x07\x12*\n%LOGICAL_PIPELINE_NODE_KIND_DETOKENIZE\x10\xf0\x07\x12%\n LOGICAL_PIPELINE_NODE_KIND_EMBED\x10\xf1\x07\x12\x35\n0LOGICAL_PIPELINE_NODE_KIND_TEMPLATE_CHAT_MESSAGE\x10\xf2\x07\x12.\n)LOGICAL_PIPELINE_NODE_KIND_ESTIMATE_DEPTH\x10\xf3\x07\x12,\n\'LOGICAL_PIPELINE_NODE_KIND_FOR_EACH_ROI\x10\xdf\x0b\x12-\n(LOGICAL_PIPELINE_NODE_KIND_VECTOR_SEARCH\x10\xe0\x0b\x12\x34\n/LOGICAL_PIPELINE_NODE_KIND_FILTER_MOVING_TRACKS\x10\xe1\x0b\x12\x33\n.LOGICAL_PIPELINE_NODE_KIND_EXTRACT_IMAGE_FRAME\x10\xe2\x0b\x12.\n)LOGICAL_PIPELINE_NODE_KIND_EXTRACT_PROMPT\x10\xe3\x0b\x12\x31\n,LOGICAL_PIPELINE_NODE_KIND_VIDEO_STREAM_SINK\x10\xd0\x0f\x12\x37\n2LOGICAL_PIPELINE_NODE_KIND_DETECTIONS_METRICS_SINK\x10\xd1\x0f\x12\x34\n/LOGICAL_PIPELINE_NODE_KIND_LATENCY_METRICS_SINK\x10\xd2\x0f\x12\x32\n-LOGICAL_PIPELINE_NODE_KIND_FRAME_METRICS_SINK\x10\xd3\x0f\x12\x30\n+LOGICAL_PIPELINE_NODE_KIND_TEXT_STREAM_SINK\x10\xd4\x0f\x12-\n(LOGICAL_PIPELINE_NODE_KIND_ENDPOINT_SINK\x10\xd5\x0f*\x98\x01\n\x0cSemanticType\x12\x19\n\x15SEMANTIC_TYPE_UNKNOWN\x10\x00\x12\x16\n\x12SEMANTIC_TYPE_TEXT\x10\x01\x12\x17\n\x13SEMANTIC_TYPE_IMAGE\x10\x02\x12\x1c\n\x18SEMANTIC_TYPE_DETECTIONS\x10\x03\x12\x1e\n\x1aSEMANTIC_TYPE_SEGMENTATION\x10\x04*\xf5\x01\n\rPipelineState\x12\x1a\n\x16PIPELINE_STATE_UNKNOWN\x10\x00\x12\x1a\n\x16PIPELINE_STATE_PENDING\x10\x01\x12\x18\n\x14PIPELINE_STATE_READY\x10\x02\x12\x1a\n\x16PIPELINE_STATE_RUNNING\x10\x03\x12\x1e\n\x1aPIPELINE_STATE_TERMINATING\x10\x04\x12\x1d\n\x19PIPELINE_STATE_TERMINATED\x10\x05\x12\x19\n\x15PIPELINE_STATE_FAILED\x10\x06\x12\x1c\n\x18PIPELINE_STATE_COMPILING\x10\x07*\xe8\x01\n\rKVCacheFormat\x12\x1b\n\x17KV_CACHE_FORMAT_UNKNOWN\x10\x00\x12:\n6KV_CACHE_FORMAT_CONTIGUOUS_SEPARATE_KV_NO_OUTPUT_SPACE\x10\x01\x12<\n8KV_CACHE_FORMAT_CONTIGUOUS_COMBINED_KV_WITH_OUTPUT_SPACE\x10\x02\x12\x1e\n\x1aKV_CACHE_FORMAT_PAGED_TLLM\x10\x03\x12 \n\x1cKV_CACHE_FORMAT_PREALLOCATED\x10\x04\x42/Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepbb\x06proto3')
 
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.api.corepb.v1.model_exec_pb2', globals())
@@ -35,6 +35,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _MODEL.fields_by_name['id']._serialized_options = b'\342\336\037\002ID'
   _MODELINFO_FILEASSETSENTRY._options = None
   _MODELINFO_FILEASSETSENTRY._serialized_options = b'8\001'
+  _VALUEDICT_VALUESENTRY._options = None
+  _VALUEDICT_VALUESENTRY._serialized_options = b'8\001'
   _VALUE_MODELREF.fields_by_name['id']._options = None
   _VALUE_MODELREF.fields_by_name['id']._serialized_options = b'\342\336\037\002ID'
   _LOGICALPIPELINE.fields_by_name['model_ids']._options = None
@@ -69,18 +71,22 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _MODELSPEC.fields_by_name['openvino_spec']._serialized_options = b'\342\336\037\014OpenVINOSpec'
   _MODELSPEC.fields_by_name['iree_spec']._options = None
   _MODELSPEC.fields_by_name['iree_spec']._serialized_options = b'\342\336\037\rIREEModelSpec'
+  _MODELSPEC.fields_by_name['habana_spec']._options = None
+  _MODELSPEC.fields_by_name['habana_spec']._serialized_options = b'\342\336\037\017HabanaModelSpec'
   _MODELSPEC.fields_by_name['runtime']._options = None
   _MODELSPEC.fields_by_name['runtime']._serialized_options = b'\030\001'
-  _KVCACHEKIND._serialized_start=13702
-  _KVCACHEKIND._serialized_end=13797
-  _POSITIONEMBEDDINGKIND._serialized_start=13800
-  _POSITIONEMBEDDINGKIND._serialized_end=13943
-  _LOGICALPIPELINENODEKIND._serialized_start=13946
-  _LOGICALPIPELINENODEKIND._serialized_end=15248
-  _PIPELINESTATE._serialized_start=15251
-  _PIPELINESTATE._serialized_end=15496
-  _KVCACHEFORMAT._serialized_start=15499
-  _KVCACHEFORMAT._serialized_end=15697
+  _KVCACHEKIND._serialized_start=14537
+  _KVCACHEKIND._serialized_end=14632
+  _POSITIONEMBEDDINGKIND._serialized_start=14635
+  _POSITIONEMBEDDINGKIND._serialized_end=14778
+  _LOGICALPIPELINENODEKIND._serialized_start=14781
+  _LOGICALPIPELINENODEKIND._serialized_end=16083
+  _SEMANTICTYPE._serialized_start=16086
+  _SEMANTICTYPE._serialized_end=16238
+  _PIPELINESTATE._serialized_start=16241
+  _PIPELINESTATE._serialized_end=16486
+  _KVCACHEFORMAT._serialized_start=16489
+  _KVCACHEFORMAT._serialized_end=16721
   _BOUNDINGBOXINFO._serialized_start=258
   _BOUNDINGBOXINFO._serialized_end=553
   _BOUNDINGBOXINFO_BOUNDINGBOXFORMAT._serialized_start=409
@@ -98,7 +104,7 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _IMAGEPREPROCESSINGSTEP_IMAGEPREPROCESSINGKIND._serialized_start=1479
   _IMAGEPREPROCESSINGSTEP_IMAGEPREPROCESSINGKIND._serialized_end=1668
   _DIMENSIONSEMANTICS._serialized_start=1671
-  _DIMENSIONSEMANTICS._serialized_end=4417
+  _DIMENSIONSEMANTICS._serialized_end=4460
   _DIMENSIONSEMANTICS_IMAGECHANNELPARAMS._serialized_start=2394
   _DIMENSIONSEMANTICS_IMAGECHANNELPARAMS._serialized_end=2675
   _DIMENSIONSEMANTICS_IMAGECHANNELPARAMS_IMAGECHANNELFORMAT._serialized_start=2525
@@ -116,99 +122,109 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _DIMENSIONSEMANTICS_REGRESSIONPARAMS._serialized_start=3606
   _DIMENSIONSEMANTICS_REGRESSIONPARAMS._serialized_end=3698
   _DIMENSIONSEMANTICS_DIMENSIONSEMANTICSKIND._serialized_start=3701
-  _DIMENSIONSEMANTICS_DIMENSIONSEMANTICSKIND._serialized_end=4417
-  _TENSORSEMANTICS._serialized_start=4420
-  _TENSORSEMANTICS._serialized_end=4781
-  _TENSORSEMANTICS_TENSORSEMANTICSKIND._serialized_start=4601
-  _TENSORSEMANTICS_TENSORSEMANTICSKIND._serialized_end=4781
-  _GENERATIONCONFIG._serialized_start=4783
-  _GENERATIONCONFIG._serialized_end=4837
-  _KVCACHECONFIG._serialized_start=4839
-  _KVCACHECONFIG._serialized_end=4913
-  _MODEL._serialized_start=4915
-  _MODEL._serialized_end=5020
-  _PACKAGEINFO._serialized_start=5022
-  _PACKAGEINFO._serialized_end=5081
-  _TRACINGMETADATA._serialized_start=5083
-  _TRACINGMETADATA._serialized_end=5174
-  _DEFAULTROPECONFIG._serialized_start=5177
-  _DEFAULTROPECONFIG._serialized_end=5324
-  _LLAMA3ROPECONFIG._serialized_start=5327
-  _LLAMA3ROPECONFIG._serialized_end=5652
-  _POSITIONEMBEDDINGCONFIG._serialized_start=5655
-  _POSITIONEMBEDDINGCONFIG._serialized_end=5932
-  _TRANSFORMERCONFIG._serialized_start=5935
-  _TRANSFORMERCONFIG._serialized_end=6065
-  _MODELINFO._serialized_start=6068
-  _MODELINFO._serialized_end=7732
-  _MODELINFO_FILEASSETSENTRY._serialized_start=7095
-  _MODELINFO_FILEASSETSENTRY._serialized_end=7173
-  _MODELINFO_MODELKIND._serialized_start=7176
-  _MODELINFO_MODELKIND._serialized_end=7392
-  _MODELINFO_MODELSTORAGEFORMAT._serialized_start=7395
-  _MODELINFO_MODELSTORAGEFORMAT._serialized_end=7732
-  _NODE._serialized_start=7735
-  _NODE._serialized_end=8031
-  _PARAMREF._serialized_start=8033
-  _PARAMREF._serialized_end=8063
-  _VALUE._serialized_start=8066
-  _VALUE._serialized_end=8621
-  _VALUE_MODELREF._serialized_start=8428
-  _VALUE_MODELREF._serialized_end=8499
-  _VALUE_LAMBDA._serialized_start=8501
-  _VALUE_LAMBDA._serialized_end=8613
-  _NODEATTRIBUTES._serialized_start=8623
-  _NODEATTRIBUTES._serialized_end=8714
-  _NODEINPUT._serialized_start=8717
-  _NODEINPUT._serialized_end=9129
-  _NODEINPUT_NODEOUTPUTREF._serialized_start=9018
-  _NODEINPUT_NODEOUTPUTREF._serialized_end=9082
-  _NODEINPUT_LAMBDAINPUTREF._serialized_start=9084
-  _NODEINPUT_LAMBDAINPUTREF._serialized_end=9120
-  _NODEOUTPUT._serialized_start=9131
-  _NODEOUTPUT._serialized_end=9163
-  _PARAM._serialized_start=9165
-  _PARAM._serialized_end=9262
-  _LOGICALPIPELINE._serialized_start=9265
-  _LOGICALPIPELINE._serialized_end=9453
-  _PIPELINEDEPLOYMENT._serialized_start=9456
-  _PIPELINEDEPLOYMENT._serialized_end=10011
-  _PIPELINEDEPLOYMENTSPEC._serialized_start=10014
-  _PIPELINEDEPLOYMENTSPEC._serialized_end=10191
-  _DEPLOYMENTSCALINGCONFIG._serialized_start=10194
-  _DEPLOYMENTSCALINGCONFIG._serialized_end=10383
-  _DEPLOYMENTSCALINGCONFIG_REPLICATEPERNODESTRATEGY._serialized_start=10345
-  _DEPLOYMENTSCALINGCONFIG_REPLICATEPERNODESTRATEGY._serialized_end=10371
-  _PIPELINEDEPLOYMENTSTATUS._serialized_start=10385
-  _PIPELINEDEPLOYMENTSTATUS._serialized_end=10498
-  _PHYSICALPIPELINE._serialized_start=10501
-  _PHYSICALPIPELINE._serialized_end=10977
-  _PHYSICALPIPELINESPEC._serialized_start=10980
-  _PHYSICALPIPELINESPEC._serialized_end=11313
-  _PHYSICALPIPELINESTATUS._serialized_start=11316
-  _PHYSICALPIPELINESTATUS._serialized_end=11836
-  _PHYSICALPIPELINESTATUS_MODELRUNTIMESPEC._serialized_start=11724
-  _PHYSICALPIPELINESTATUS_MODELRUNTIMESPEC._serialized_end=11836
-  _FILERESOURCE._serialized_start=11839
-  _FILERESOURCE._serialized_end=11971
-  _EXECUTIONSPEC._serialized_start=11974
-  _EXECUTIONSPEC._serialized_end=12113
-  _NAMEDASSET._serialized_start=12115
-  _NAMEDASSET._serialized_end=12207
-  _MODELSPEC._serialized_start=12210
-  _MODELSPEC._serialized_end=12885
-  _TENSORRTMODELSPEC._serialized_start=12888
-  _TENSORRTMODELSPEC._serialized_end=13142
-  _TENSORRTOPTIMIZATIONPROFILE._serialized_start=13144
-  _TENSORRTOPTIMIZATIONPROFILE._serialized_end=13271
-  _TENSORRTTENSORSHAPERANGE._serialized_start=13273
-  _TENSORRTTENSORSHAPERANGE._serialized_end=13392
-  _TENSORRTMEMPOOLLIMITS._serialized_start=13394
-  _TENSORRTMEMPOOLLIMITS._serialized_end=13447
-  _OPENVINOMODELSPEC._serialized_start=13450
-  _OPENVINOMODELSPEC._serialized_end=13621
-  _OPENVINOMODELSPEC_TENSORSHAPE._serialized_start=13590
-  _OPENVINOMODELSPEC_TENSORSHAPE._serialized_end=13621
-  _IREEMODELSPEC._serialized_start=13623
-  _IREEMODELSPEC._serialized_end=13700
+  _DIMENSIONSEMANTICS_DIMENSIONSEMANTICSKIND._serialized_end=4460
+  _TENSORSEMANTICS._serialized_start=4463
+  _TENSORSEMANTICS._serialized_end=4824
+  _TENSORSEMANTICS_TENSORSEMANTICSKIND._serialized_start=4644
+  _TENSORSEMANTICS_TENSORSEMANTICSKIND._serialized_end=4824
+  _GENERATIONCONFIG._serialized_start=4826
+  _GENERATIONCONFIG._serialized_end=4880
+  _KVCACHECONFIG._serialized_start=4882
+  _KVCACHECONFIG._serialized_end=4956
+  _MODEL._serialized_start=4958
+  _MODEL._serialized_end=5063
+  _PACKAGEINFO._serialized_start=5065
+  _PACKAGEINFO._serialized_end=5124
+  _TRACINGMETADATA._serialized_start=5126
+  _TRACINGMETADATA._serialized_end=5217
+  _DEFAULTROPECONFIG._serialized_start=5220
+  _DEFAULTROPECONFIG._serialized_end=5367
+  _LLAMA3ROPECONFIG._serialized_start=5370
+  _LLAMA3ROPECONFIG._serialized_end=5695
+  _POSITIONEMBEDDINGCONFIG._serialized_start=5698
+  _POSITIONEMBEDDINGCONFIG._serialized_end=5975
+  _TRANSFORMERCONFIG._serialized_start=5978
+  _TRANSFORMERCONFIG._serialized_end=6108
+  _MODELINFO._serialized_start=6111
+  _MODELINFO._serialized_end=7835
+  _MODELINFO_FILEASSETSENTRY._serialized_start=7138
+  _MODELINFO_FILEASSETSENTRY._serialized_end=7216
+  _MODELINFO_MODELKIND._serialized_start=7219
+  _MODELINFO_MODELKIND._serialized_end=7458
+  _MODELINFO_MODELSTORAGEFORMAT._serialized_start=7461
+  _MODELINFO_MODELSTORAGEFORMAT._serialized_end=7835
+  _MODELINFOLIST._serialized_start=7837
+  _MODELINFOLIST._serialized_end=7913
+  _NODE._serialized_start=7916
+  _NODE._serialized_end=8212
+  _PARAMREF._serialized_start=8214
+  _PARAMREF._serialized_end=8244
+  _VALUELIST._serialized_start=8246
+  _VALUELIST._serialized_end=8314
+  _VALUEDICT._serialized_start=8317
+  _VALUEDICT._serialized_end=8493
+  _VALUEDICT_VALUESENTRY._serialized_start=8403
+  _VALUEDICT_VALUESENTRY._serialized_end=8493
+  _VALUE._serialized_start=8496
+  _VALUE._serialized_end=9270
+  _VALUE_MODELREF._serialized_start=9077
+  _VALUE_MODELREF._serialized_end=9148
+  _VALUE_LAMBDA._serialized_start=9150
+  _VALUE_LAMBDA._serialized_end=9262
+  _NODEATTRIBUTES._serialized_start=9272
+  _NODEATTRIBUTES._serialized_end=9363
+  _NODEINPUT._serialized_start=9366
+  _NODEINPUT._serialized_end=9778
+  _NODEINPUT_NODEOUTPUTREF._serialized_start=9667
+  _NODEINPUT_NODEOUTPUTREF._serialized_end=9731
+  _NODEINPUT_LAMBDAINPUTREF._serialized_start=9733
+  _NODEINPUT_LAMBDAINPUTREF._serialized_end=9769
+  _NODEOUTPUT._serialized_start=9780
+  _NODEOUTPUT._serialized_end=9812
+  _PARAM._serialized_start=9814
+  _PARAM._serialized_end=9911
+  _LOGICALPIPELINE._serialized_start=9914
+  _LOGICALPIPELINE._serialized_end=10102
+  _PIPELINEDEPLOYMENT._serialized_start=10105
+  _PIPELINEDEPLOYMENT._serialized_end=10660
+  _PIPELINEDEPLOYMENTSPEC._serialized_start=10663
+  _PIPELINEDEPLOYMENTSPEC._serialized_end=10840
+  _DEPLOYMENTSCALINGCONFIG._serialized_start=10843
+  _DEPLOYMENTSCALINGCONFIG._serialized_end=11032
+  _DEPLOYMENTSCALINGCONFIG_REPLICATEPERNODESTRATEGY._serialized_start=10994
+  _DEPLOYMENTSCALINGCONFIG_REPLICATEPERNODESTRATEGY._serialized_end=11020
+  _PIPELINEDEPLOYMENTSTATUS._serialized_start=11034
+  _PIPELINEDEPLOYMENTSTATUS._serialized_end=11147
+  _PHYSICALPIPELINE._serialized_start=11150
+  _PHYSICALPIPELINE._serialized_end=11626
+  _PHYSICALPIPELINESPEC._serialized_start=11629
+  _PHYSICALPIPELINESPEC._serialized_end=11962
+  _PHYSICALPIPELINESTATUS._serialized_start=11965
+  _PHYSICALPIPELINESTATUS._serialized_end=12485
+  _PHYSICALPIPELINESTATUS_MODELRUNTIMESPEC._serialized_start=12373
+  _PHYSICALPIPELINESTATUS_MODELRUNTIMESPEC._serialized_end=12485
+  _FILERESOURCE._serialized_start=12488
+  _FILERESOURCE._serialized_end=12620
+  _EXECUTIONSPEC._serialized_start=12623
+  _EXECUTIONSPEC._serialized_end=12762
+  _NAMEDASSET._serialized_start=12764
+  _NAMEDASSET._serialized_end=12856
+  _MODELSPEC._serialized_start=12859
+  _MODELSPEC._serialized_end=13672
+  _TENSORRTMODELSPEC._serialized_start=13675
+  _TENSORRTMODELSPEC._serialized_end=13929
+  _TENSORRTOPTIMIZATIONPROFILE._serialized_start=13931
+  _TENSORRTOPTIMIZATIONPROFILE._serialized_end=14058
+  _TENSORRTTENSORSHAPERANGE._serialized_start=14060
+  _TENSORRTTENSORSHAPERANGE._serialized_end=14179
+  _TENSORRTMEMPOOLLIMITS._serialized_start=14181
+  _TENSORRTMEMPOOLLIMITS._serialized_end=14234
+  _OPENVINOMODELSPEC._serialized_start=14237
+  _OPENVINOMODELSPEC._serialized_end=14408
+  _OPENVINOMODELSPEC_TENSORSHAPE._serialized_start=14377
+  _OPENVINOMODELSPEC_TENSORSHAPE._serialized_end=14408
+  _IREEMODELSPEC._serialized_start=14410
+  _IREEMODELSPEC._serialized_end=14487
+  _HABANAMODELSPEC._serialized_start=14489
+  _HABANAMODELSPEC._serialized_end=14535
 # @@protoc_insertion_point(module_scope)

gml/proto/src/controlplane/compiler/cpb/v1/cpb_pb2.py

@@ -20,7 +20,7 @@ from gml.proto.src.common.typespb import uuid_pb2 as src_dot_common_dot_typespb_
 from gml.proto.gogoproto import gogo_pb2 as gogoproto_dot_gogo__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n*src/controlplane/compiler/cpb/v1/cpb.proto\x12%gml.internal.controlplane.compiler.v1\x1a\"src/api/corepb/v1/model_exec.proto\x1a)src/api/corepb/v1/compiled_pipeline.proto\x1a$src/api/corepb/v1/controlplane.proto\x1a\x1fsrc/api/corepb/v1/cp_edge.proto\x1a\"src/api/corepb/v1/gem_config.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\x14gogoproto/gogo.proto\"\xa7\x04\n\x0e\x43ompileRequest\x12V\n\x13logical_pipeline_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x15\xe2\xde\x1f\x11LogicalPipelineIDR\x11logicalPipelineId\x12T\n\x10logical_pipeline\x18\x02 \x01(\x0b\x32).gml.internal.api.core.v1.LogicalPipelineR\x0flogicalPipeline\x12\x61\n\x13\x64\x65vice_capabilities\x18\x03 \x01(\x0b\x32,.gml.internal.api.core.v1.DeviceCapabilitiesB\x02\x18\x01R\x12\x64\x65viceCapabilities\x12\x37\n\x06models\x18\x04 \x03(\x0b\x32\x1f.gml.internal.api.core.v1.ModelR\x06models\x12\x46\n\ngem_config\x18\x05 \x01(\x0b\x32#.gml.internal.api.core.v1.GEMConfigB\x02\x18\x01R\tgemConfig\x12:\n\x07\x64\x65vices\x18\x06 \x03(\x0b\x32 .gml.internal.api.core.v1.DeviceR\x07\x64\x65vices\x12G\n\x0c\x64\x65vice_links\x18\x07 \x03(\x0b\x32$.gml.internal.api.core.v1.DeviceLinkR\x0b\x64\x65viceLinks\"\x92\x03\n\x0f\x43ompileResponse\x12H\n\texec_spec\x18\x01 \x01(\x0b\x32\'.gml.internal.api.core.v1.ExecutionSpecB\x02\x18\x01R\x08\x65xecSpec\x12h\n\nexec_specs\x18\x02 \x03(\x0b\x32\x45.gml.internal.controlplane.compiler.v1.CompileResponse.ExecSpecsEntryB\x02\x18\x01R\texecSpecs\x12\x64\n\x16\x63ompiled_pipeline_spec\x18\x03 \x01(\x0b\x32..gml.internal.api.core.v1.CompiledPipelineSpecR\x14\x63ompiledPipelineSpec\x1a\x65\n\x0e\x45xecSpecsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12=\n\x05value\x18\x02 \x01(\x0b\x32\'.gml.internal.api.core.v1.ExecutionSpecR\x05value:\x02\x38\x01\x32\x8b\x01\n\x0f\x43ompilerService\x12x\n\x07\x43ompile\x12\x35.gml.internal.controlplane.compiler.v1.CompileRequest\x1a\x36.gml.internal.controlplane.compiler.v1.CompileResponseB;Z9gimletlabs.ai/gimlet/src/controlplane/compiler/cpb/v1;cpbb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n*src/controlplane/compiler/cpb/v1/cpb.proto\x12%gml.internal.controlplane.compiler.v1\x1a\"src/api/corepb/v1/model_exec.proto\x1a)src/api/corepb/v1/compiled_pipeline.proto\x1a$src/api/corepb/v1/controlplane.proto\x1a\x1fsrc/api/corepb/v1/cp_edge.proto\x1a\"src/api/corepb/v1/gem_config.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\x14gogoproto/gogo.proto\"\xe9\x04\n\x0e\x43ompileRequest\x12V\n\x13logical_pipeline_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x15\xe2\xde\x1f\x11LogicalPipelineIDR\x11logicalPipelineId\x12T\n\x10logical_pipeline\x18\x02 \x01(\x0b\x32).gml.internal.api.core.v1.LogicalPipelineR\x0flogicalPipeline\x12\x61\n\x13\x64\x65vice_capabilities\x18\x03 \x01(\x0b\x32,.gml.internal.api.core.v1.DeviceCapabilitiesB\x02\x18\x01R\x12\x64\x65viceCapabilities\x12\x37\n\x06models\x18\x04 \x03(\x0b\x32\x1f.gml.internal.api.core.v1.ModelR\x06models\x12\x46\n\ngem_config\x18\x05 \x01(\x0b\x32#.gml.internal.api.core.v1.GEMConfigB\x02\x18\x01R\tgemConfig\x12:\n\x07\x64\x65vices\x18\x06 \x03(\x0b\x32 .gml.internal.api.core.v1.DeviceR\x07\x64\x65vices\x12G\n\x0c\x64\x65vice_links\x18\x07 \x03(\x0b\x32$.gml.internal.api.core.v1.DeviceLinkR\x0b\x64\x65viceLinks\x12@\n\x0b\x65ndpoint_id\x18\x08 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0e\xe2\xde\x1f\nEndpointIDR\nendpointId\"\x92\x03\n\x0f\x43ompileResponse\x12H\n\texec_spec\x18\x01 \x01(\x0b\x32\'.gml.internal.api.core.v1.ExecutionSpecB\x02\x18\x01R\x08\x65xecSpec\x12h\n\nexec_specs\x18\x02 \x03(\x0b\x32\x45.gml.internal.controlplane.compiler.v1.CompileResponse.ExecSpecsEntryB\x02\x18\x01R\texecSpecs\x12\x64\n\x16\x63ompiled_pipeline_spec\x18\x03 \x01(\x0b\x32..gml.internal.api.core.v1.CompiledPipelineSpecR\x14\x63ompiledPipelineSpec\x1a\x65\n\x0e\x45xecSpecsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12=\n\x05value\x18\x02 \x01(\x0b\x32\'.gml.internal.api.core.v1.ExecutionSpecR\x05value:\x02\x38\x01\x32\x8b\x01\n\x0f\x43ompilerService\x12x\n\x07\x43ompile\x12\x35.gml.internal.controlplane.compiler.v1.CompileRequest\x1a\x36.gml.internal.controlplane.compiler.v1.CompileResponseB;Z9gimletlabs.ai/gimlet/src/controlplane/compiler/cpb/v1;cpbb\x06proto3')
 
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.controlplane.compiler.cpb.v1.cpb_pb2', globals())
@@ -34,6 +34,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _COMPILEREQUEST.fields_by_name['device_capabilities']._serialized_options = b'\030\001'
   _COMPILEREQUEST.fields_by_name['gem_config']._options = None
   _COMPILEREQUEST.fields_by_name['gem_config']._serialized_options = b'\030\001'
+  _COMPILEREQUEST.fields_by_name['endpoint_id']._options = None
+  _COMPILEREQUEST.fields_by_name['endpoint_id']._serialized_options = b'\342\336\037\nEndpointID'
   _COMPILERESPONSE_EXECSPECSENTRY._options = None
   _COMPILERESPONSE_EXECSPECSENTRY._serialized_options = b'8\001'
   _COMPILERESPONSE.fields_by_name['exec_spec']._options = None
@@ -41,11 +43,11 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _COMPILERESPONSE.fields_by_name['exec_specs']._options = None
   _COMPILERESPONSE.fields_by_name['exec_specs']._serialized_options = b'\030\001'
   _COMPILEREQUEST._serialized_start=325
-  _COMPILEREQUEST._serialized_end=876
-  _COMPILERESPONSE._serialized_start=879
-  _COMPILERESPONSE._serialized_end=1281
-  _COMPILERESPONSE_EXECSPECSENTRY._serialized_start=1180
-  _COMPILERESPONSE_EXECSPECSENTRY._serialized_end=1281
-  _COMPILERSERVICE._serialized_start=1284
-  _COMPILERSERVICE._serialized_end=1423
+  _COMPILEREQUEST._serialized_end=942
+  _COMPILERESPONSE._serialized_start=945
+  _COMPILERESPONSE._serialized_end=1347
+  _COMPILERESPONSE_EXECSPECSENTRY._serialized_start=1246
+  _COMPILERESPONSE_EXECSPECSENTRY._serialized_end=1347
+  _COMPILERSERVICE._serialized_start=1350
+  _COMPILERSERVICE._serialized_end=1489
 # @@protoc_insertion_point(module_scope)

gml/tensor.py

@@ -278,6 +278,13 @@ class EmbeddingDimension(DimensionSemantics):
         )
 
 
+class PositionIDsDimension(DimensionSemantics):
+    def to_proto(self) -> modelexecpb.DimensionSemantics:
+        return modelexecpb.DimensionSemantics(
+            kind=modelexecpb.DimensionSemantics.DIMENSION_SEMANTICS_KIND_POSITION_IDS,
+        )
+
+
 class TensorSemantics:
     def __init__(
         self,