From 076424888315aaf2f8918b1b6903d10cecad6236 Mon Sep 17 00:00:00 2001
From: Siddharth Venkatesan <siddvenk@umich.edu>
Date: Wed, 29 Mar 2023 09:03:43 -0700
Subject: [PATCH] [fix] update allowed instance types for DJLModels

---
 src/sagemaker/djl_inference/defaults.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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