Skip to content

Commit 0764248

Browse files
siddvenktrajanikant
authored andcommitted
[fix] update allowed instance types for DJLModels
1 parent 19115a2 commit 0764248

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/sagemaker/djl_inference/defaults.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,8 @@
4747
"ml.g4dn",
4848
"ml.g5",
4949
"ml.p3",
50-
"ml.p4",
50+
"ml.p3dn"
51+
"ml.p4d",
52+
"ml.p4de"
5153
"local_gpu",
5254
}

0 commit comments

Comments
 (0)