Skip to content

Commit 9f103b4

Browse files
committed
[fix] update allowed instance types for DJLModels
1 parent 0473a8e commit 9f103b4

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/sagemaker/djl_inference/defaults.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,8 @@
4545
"ml.g4dn",
4646
"ml.g5",
4747
"ml.p3",
48-
"ml.p4",
48+
"ml.p3dn"
49+
"ml.p4d",
50+
"ml.p4de"
4951
"local_gpu",
5052
}

0 commit comments

Comments
 (0)