Mhh I don’t think it is available off the shelf, but let’s wait for Robert.
Pobably a custom component is required, something like:

  

      github.com
  

  
    GoogleCloudPlatform/mlops-with-vertex-ai/blob/main/src/tfx_pipelines/components.py#L73-L116 5



      
      
          @component
          def vertex_model_uploader(
              project: Parameter[str],
              region: Parameter[str],
              model_display_name: Parameter[str],
              pushed_model_location: Parameter[str],
              serving_image_uri: Parameter[str],
              explanation_config: Parameter[str],
              uploaded_model: OutputArtifact[UploadedModel],
          ):
          
          
    vertex_ai.init(project=project, location=region)
          
          
    pushed_model_dir = os.path.join(
                  pushed_model_location, tf.io.gfile.listdir(pushed_model_location)[-1]
              )
          
          
    logging.info(f"Model registry location: {pushed_model_dir}")
          
          
    try:
      
    
  

  This file has been truncated. show original