Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions CONTRIBUTING.md
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@ information to effectively respond to your bug report or contribution.
* [Run the Unit Tests](#run-the-unit-tests)
* [Run the Integration Tests](#run-the-integration-tests)
* [Make and Test Your Change](#make-and-test-your-change)
* [Lint Your Change](#lint-your-change)
* [Commit Your Change](#commit-your-change)
* [Send a Pull Request](#send-a-pull-request)
* [Documentation Guidelines](#documentation-guidelines)
Expand Down Expand Up @@ -117,6 +118,13 @@ If you are writing or modifying a test that creates a SageMaker job (training, t
1. If your changes include documentation changes, please see the [Documentation Guidelines](#documentation-guidelines).
1. If you include integration tests, do not mark them as canaries if they will not run in all regions.

### Lint Your Change

Before submitting, ensure your code meets our quality and style guidelines. Run:
```shell
tox -e flake8,pylint,docstyle,black-check,twine --parallel all
```
Address any errors or warnings before opening a pull request.

### Commit Your Change

Expand Down
90 changes: 81 additions & 9 deletions src/sagemaker/image_uri_config/djl-deepspeed.json
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,15 @@
"us-gov-west-1": "442386744353",
"us-west-1": "763104351884",
"us-west-2": "763104351884",
"ca-west-1": "204538143572"
"ca-west-1": "204538143572",
"eu-central-2": "380420809688",
"ap-south-2": "772153158452",
"me-central-1": "914824155844",
"eu-south-2": "503227376785",
"ap-southeast-7": "590183813437",
"ap-southeast-4": "457447274322",
"ap-southeast-5": "550225433462",
"mx-central-1": "637423239942"
},
"repository": "djl-inference",
"tag_prefix": "0.27.0-deepspeed0.12.6-cu121"
Expand Down Expand Up @@ -66,7 +74,15 @@
"us-gov-west-1": "442386744353",
"us-west-1": "763104351884",
"us-west-2": "763104351884",
"ca-west-1": "204538143572"
"ca-west-1": "204538143572",
"eu-central-2": "380420809688",
"ap-south-2": "772153158452",
"me-central-1": "914824155844",
"eu-south-2": "503227376785",
"ap-southeast-7": "590183813437",
"ap-southeast-4": "457447274322",
"ap-southeast-5": "550225433462",
"mx-central-1": "637423239942"
},
"repository": "djl-inference",
"tag_prefix": "0.26.0-deepspeed0.12.6-cu121"
Expand Down Expand Up @@ -100,7 +116,15 @@
"us-gov-west-1": "442386744353",
"us-west-1": "763104351884",
"us-west-2": "763104351884",
"ca-west-1": "204538143572"
"ca-west-1": "204538143572",
"eu-central-2": "380420809688",
"ap-south-2": "772153158452",
"me-central-1": "914824155844",
"eu-south-2": "503227376785",
"ap-southeast-7": "590183813437",
"ap-southeast-4": "457447274322",
"ap-southeast-5": "550225433462",
"mx-central-1": "637423239942"
},
"repository": "djl-inference",
"tag_prefix": "0.25.0-deepspeed0.11.0-cu118"
Expand Down Expand Up @@ -134,7 +158,15 @@
"us-gov-west-1": "442386744353",
"us-west-1": "763104351884",
"us-west-2": "763104351884",
"ca-west-1": "204538143572"
"ca-west-1": "204538143572",
"eu-central-2": "380420809688",
"ap-south-2": "772153158452",
"me-central-1": "914824155844",
"eu-south-2": "503227376785",
"ap-southeast-7": "590183813437",
"ap-southeast-4": "457447274322",
"ap-southeast-5": "550225433462",
"mx-central-1": "637423239942"
},
"repository": "djl-inference",
"tag_prefix": "0.24.0-deepspeed0.10.0-cu118"
Expand Down Expand Up @@ -168,7 +200,15 @@
"us-gov-west-1": "442386744353",
"us-west-1": "763104351884",
"us-west-2": "763104351884",
"ca-west-1": "204538143572"
"ca-west-1": "204538143572",
"eu-central-2": "380420809688",
"ap-south-2": "772153158452",
"me-central-1": "914824155844",
"eu-south-2": "503227376785",
"ap-southeast-7": "590183813437",
"ap-southeast-4": "457447274322",
"ap-southeast-5": "550225433462",
"mx-central-1": "637423239942"
},
"repository": "djl-inference",
"tag_prefix": "0.23.0-deepspeed0.9.5-cu118"
Expand Down Expand Up @@ -202,7 +242,15 @@
"us-gov-west-1": "442386744353",
"us-west-1": "763104351884",
"us-west-2": "763104351884",
"ca-west-1": "204538143572"
"ca-west-1": "204538143572",
"eu-central-2": "380420809688",
"ap-south-2": "772153158452",
"me-central-1": "914824155844",
"eu-south-2": "503227376785",
"ap-southeast-7": "590183813437",
"ap-southeast-4": "457447274322",
"ap-southeast-5": "550225433462",
"mx-central-1": "637423239942"
},
"repository": "djl-inference",
"tag_prefix": "0.22.1-deepspeed0.9.2-cu118"
Expand Down Expand Up @@ -236,7 +284,15 @@
"us-gov-west-1": "442386744353",
"us-west-1": "763104351884",
"us-west-2": "763104351884",
"ca-west-1": "204538143572"
"ca-west-1": "204538143572",
"eu-central-2": "380420809688",
"ap-south-2": "772153158452",
"me-central-1": "914824155844",
"eu-south-2": "503227376785",
"ap-southeast-7": "590183813437",
"ap-southeast-4": "457447274322",
"ap-southeast-5": "550225433462",
"mx-central-1": "637423239942"
},
"repository": "djl-inference",
"tag_prefix": "0.21.0-deepspeed0.8.3-cu117"
Expand Down Expand Up @@ -270,7 +326,15 @@
"us-gov-west-1": "442386744353",
"us-west-1": "763104351884",
"us-west-2": "763104351884",
"ca-west-1": "204538143572"
"ca-west-1": "204538143572",
"eu-central-2": "380420809688",
"ap-south-2": "772153158452",
"me-central-1": "914824155844",
"eu-south-2": "503227376785",
"ap-southeast-7": "590183813437",
"ap-southeast-4": "457447274322",
"ap-southeast-5": "550225433462",
"mx-central-1": "637423239942"
},
"repository": "djl-inference",
"tag_prefix": "0.20.0-deepspeed0.7.5-cu116"
Expand Down Expand Up @@ -304,7 +368,15 @@
"us-gov-west-1": "442386744353",
"us-west-1": "763104351884",
"us-west-2": "763104351884",
"ca-west-1": "204538143572"
"ca-west-1": "204538143572",
"eu-central-2": "380420809688",
"ap-south-2": "772153158452",
"me-central-1": "914824155844",
"eu-south-2": "503227376785",
"ap-southeast-7": "590183813437",
"ap-southeast-4": "457447274322",
"ap-southeast-5": "550225433462",
"mx-central-1": "637423239942"
},
"repository": "djl-inference",
"tag_prefix": "0.19.0-deepspeed0.7.3-cu113"
Expand Down
42 changes: 37 additions & 5 deletions src/sagemaker/image_uri_config/djl-fastertransformer.json
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,15 @@
"us-east-2": "763104351884",
"us-west-1": "763104351884",
"us-west-2": "763104351884",
"ca-west-1": "204538143572"
"ca-west-1": "204538143572",
"eu-central-2": "380420809688",
"ap-south-2": "772153158452",
"me-central-1": "914824155844",
"eu-south-2": "503227376785",
"ap-southeast-7": "590183813437",
"ap-southeast-4": "457447274322",
"ap-southeast-5": "550225433462",
"mx-central-1": "637423239942"
},
"repository": "djl-inference",
"tag_prefix": "0.24.0-fastertransformer5.3.0-cu118"
Expand Down Expand Up @@ -62,7 +70,15 @@
"us-east-2": "763104351884",
"us-west-1": "763104351884",
"us-west-2": "763104351884",
"ca-west-1": "204538143572"
"ca-west-1": "204538143572",
"eu-central-2": "380420809688",
"ap-south-2": "772153158452",
"me-central-1": "914824155844",
"eu-south-2": "503227376785",
"ap-southeast-7": "590183813437",
"ap-southeast-4": "457447274322",
"ap-southeast-5": "550225433462",
"mx-central-1": "637423239942"
},
"repository": "djl-inference",
"tag_prefix": "0.23.0-fastertransformer5.3.0-cu118"
Expand Down Expand Up @@ -94,7 +110,15 @@
"us-east-2": "763104351884",
"us-west-1": "763104351884",
"us-west-2": "763104351884",
"ca-west-1": "204538143572"
"ca-west-1": "204538143572",
"eu-central-2": "380420809688",
"ap-south-2": "772153158452",
"me-central-1": "914824155844",
"eu-south-2": "503227376785",
"ap-southeast-7": "590183813437",
"ap-southeast-4": "457447274322",
"ap-southeast-5": "550225433462",
"mx-central-1": "637423239942"
},
"repository": "djl-inference",
"tag_prefix": "0.22.1-fastertransformer5.3.0-cu118"
Expand Down Expand Up @@ -126,10 +150,18 @@
"us-east-2": "763104351884",
"us-west-1": "763104351884",
"us-west-2": "763104351884",
"ca-west-1": "204538143572"
"ca-west-1": "204538143572",
"eu-central-2": "380420809688",
"ap-south-2": "772153158452",
"me-central-1": "914824155844",
"eu-south-2": "503227376785",
"ap-southeast-7": "590183813437",
"ap-southeast-4": "457447274322",
"ap-southeast-5": "550225433462",
"mx-central-1": "637423239942"
},
"repository": "djl-inference",
"tag_prefix": "0.21.0-fastertransformer5.3.0-cu117"
}
}
}
}
27 changes: 24 additions & 3 deletions src/sagemaker/image_uri_config/djl-lmi.json
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,14 @@
"us-gov-west-1": "442386744353",
"us-west-1": "763104351884",
"us-west-2": "763104351884",
"ca-west-1": "204538143572"
"ca-west-1": "204538143572",
"eu-central-2": "380420809688",
"ap-south-2": "772153158452",
"eu-south-2": "503227376785",
"ap-southeast-7": "590183813437",
"ap-southeast-4": "457447274322",
"ap-southeast-5": "550225433462",
"mx-central-1": "637423239942"
},
"repository": "djl-inference",
"tag_prefix": "0.30.0-lmi12.0.0-cu124"
Expand Down Expand Up @@ -71,7 +78,14 @@
"us-gov-west-1": "442386744353",
"us-west-1": "763104351884",
"us-west-2": "763104351884",
"ca-west-1": "204538143572"
"ca-west-1": "204538143572",
"eu-central-2": "380420809688",
"ap-south-2": "772153158452",
"eu-south-2": "503227376785",
"ap-southeast-7": "590183813437",
"ap-southeast-4": "457447274322",
"ap-southeast-5": "550225433462",
"mx-central-1": "637423239942"
},
"repository": "djl-inference",
"tag_prefix": "0.29.0-lmi11.0.0-cu124"
Expand Down Expand Up @@ -106,7 +120,14 @@
"us-gov-west-1": "442386744353",
"us-west-1": "763104351884",
"us-west-2": "763104351884",
"ca-west-1": "204538143572"
"ca-west-1": "204538143572",
"eu-central-2": "380420809688",
"ap-south-2": "772153158452",
"eu-south-2": "503227376785",
"ap-southeast-7": "590183813437",
"ap-southeast-4": "457447274322",
"ap-southeast-5": "550225433462",
"mx-central-1": "637423239942"
},
"repository": "djl-inference",
"tag_prefix": "0.28.0-lmi10.0.0-cu124"
Expand Down
Loading