release_images.yml

---
release_images:
  1:
    framework: "pytorch"
    version: "1.11.0"
    arch_type: "x86"
    training:
      device_types: ["neuron"]
      python_versions: ["py38"]
      os_version: "ubuntu20.04"
      neuron_sdk_version: "sdk2.4.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: "djl"
    version: "0.19.0"
    arch_type: "x86"
    inference:
      device_types: [ "gpu" ]
      python_versions: [ "py38" ]
      os_version: "ubuntu20.04"
      deepspeed_version: "0.7.3"
      cuda_version: "cu113"
      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.
  3:
    framework: "huggingface_pytorch_trcomp"
    version: "1.11.0"
    hf_transformers: "4.21.1"
    arch_type: "x86"
    training:
      device_types: [ "gpu" ]
      python_versions: [ "py38" ]
      os_version: "ubuntu20.04"
      cuda_version: "cu113"
      example: False
      disable_sm_tag: False  # [Default: False] This option is not used by Example images
      force_release: False