diff --git a/src/sagemaker/djl_inference/defaults.py b/src/sagemaker/djl_inference/defaults.py index 04cdb5844e..125fd7b676 100644 --- a/src/sagemaker/djl_inference/defaults.py +++ b/src/sagemaker/djl_inference/defaults.py @@ -47,6 +47,9 @@ "ml.g4dn", "ml.g5", "ml.p3", + "ml.p3dn", "ml.p4", + "ml.p4d", + "ml.p4de", "local_gpu", }