docker-compose.yaml

# Local development demo/example compose file
# Composes:
# - a mlmd-server connected to a SQLite file in test/config/ml-metadata/metadata.sqlite.db,
# - a model-registry server connected to the mlmd-server.
version: '3'
services:
  mlmd-server:
    image: gcr.io/tfx-oss-public/ml_metadata_store_server:1.14.0
    container_name: mlmd-server
    ports:
      - "9090:8080"
    environment:
      - METADATA_STORE_SERVER_CONFIG_FILE=/tmp/shared/conn_config.pb
    volumes:
      - ./test/config/ml-metadata:/tmp/shared
  model-registry:
    image: ghcr.io/kubeflow/model-registry/server:latest
    command: ["proxy", "--hostname", "0.0.0.0", "--mlmd-hostname", "mlmd-server", "--mlmd-port", "8080", "--datastore-type", "mlmd"]
    container_name: model-registry
    ports:
      - "8080:8080"
    depends_on:
      - mlmd-server