---
release_images:
  1:
    framework: "djl"
    version: "0.23.0"
    arch_type: "x86"
    inference:
      device_types: [ "neuronx" ]
      python_versions: [ "py38" ]
      os_version: "ubuntu20.04"
      neuron_sdk_version: "sdk2.12.0"
      example: False        # [Default: False] Set to True to denote that this image is an Example image
      disable_sm_tag: True # [Default: False] Set to True to prevent SageMaker Abbreviated Tags from being attached
      # to images being published.
      force_release: False  # [Default: False] Set to True to force images to be published even if the same image
      # has already been published. Re-released image will have minor version incremented by 1.
  2:
    framework: "huggingface_pytorch"
    version: "2.0.0"
    arch_type: "x86"
    hf_transformers: "4.28.1"
    inference:
      device_types: [ "cpu", "gpu" ]
      python_versions: [ "py310" ]
      os_version: "ubuntu20.04"
      cuda_version: "cu118"
      example: False        # [Default: False] Set to True to denote that this image is an Example image
      disable_sm_tag: False # [Default: False] Set to True to prevent SageMaker Abbreviated Tags from being attached
      # to images being published.
      force_release: False  # [Default: False] Set to True to force images to be published even if the same image
      # has already been published. Re-released image will have minor version incremented by 1.
  3:
    framework: "autogluon"
    version: "0.7.0"
    arch_type: "x86"
    inference:
      device_types: ["cpu", "gpu"]
      python_versions: ["py39"]
      os_version: "ubuntu20.04"
      cuda_version: "cu117"
      example: False
      disable_sm_tag: False  # [Default: False] This option is not used by Example images
      force_release: False