Skip to content

Commit e7c1db6

Browse files
author
ytlee93
committed
Add ml.p5.4xlarge instance type support
1 parent 160cd80 commit e7c1db6

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/sagemaker/hyperpod/training/quota_allocation_util.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
"ml.p4d.24xlarge": {"cpu": 96, "gpu": 8, "trainium": 0, "memory": 1152},
2626
"ml.p4de.24xlarge": {"cpu": 96, "gpu": 8, "trainium": 0, "memory": 1152},
2727
"ml.p5.48xlarge": {"cpu": 192, "gpu": 8, "trainium": 0, "memory": 2048},
28+
"ml.p5.4xlarge": {"cpu": 16, "gpu": 1, "trainium": 0, "memory": 256},
2829
"ml.trn1.32xlarge": {"cpu": 128, "gpu": 0, "trainium": 16, "memory": 512},
2930
"ml.trn1n.32xlarge": {"cpu": 128, "gpu": 0, "trainium": 16, "memory": 512},
3031
"ml.g5.xlarge": {"cpu": 4, "gpu": 1, "trainium": 0, "memory": 16},

0 commit comments

Comments
 (0)