diff --git a/stable/aws-efa-k8s-device-plugin/Chart.yaml b/stable/aws-efa-k8s-device-plugin/Chart.yaml
index 683598ee..2abfc4b6 100644
--- a/stable/aws-efa-k8s-device-plugin/Chart.yaml
+++ b/stable/aws-efa-k8s-device-plugin/Chart.yaml
@@ -1,7 +1,7 @@
 apiVersion: v1
 name: aws-efa-k8s-device-plugin
 description: A Helm chart for EFA device plugin.
-version: v0.5.5
+version: v0.5.6
 appVersion: "v0.5.4"
 home: https://github.com/aws/eks-charts
 icon: https://raw.githubusercontent.com/aws/eks-charts/master/docs/logo/aws.png
diff --git a/stable/aws-efa-k8s-device-plugin/values.yaml b/stable/aws-efa-k8s-device-plugin/values.yaml
index ece9a2aa..cd1fe0ae 100644
--- a/stable/aws-efa-k8s-device-plugin/values.yaml
+++ b/stable/aws-efa-k8s-device-plugin/values.yaml
@@ -36,6 +36,8 @@ supportedInstanceLabels: # EFA supported instances: https://docs.aws.amazon.com/
     - m7i.metal-48xl
     - c5n.9xlarge
     - c5n.18xlarge
+    - ml.c5n.9xlarge
+    - ml.c5n.18xlarge
     - c5n.metal
     - c6a.48xlarge
     - c6a.metal
@@ -101,6 +103,11 @@ supportedInstanceLabels: # EFA supported instances: https://docs.aws.amazon.com/
     - g5.16xlarge
     - g5.24xlarge
     - g5.48xlarge
+    - ml.g5.8xlarge
+    - ml.g5.12xlarge
+    - ml.g5.16xlarge
+    - ml.g5.24xlarge
+    - ml.g5.48xlarge
     - g6.8xlarge
     - g6.12xlarge
     - g6.16xlarge
@@ -116,10 +123,16 @@ supportedInstanceLabels: # EFA supported instances: https://docs.aws.amazon.com/
     - p3dn.24xlarge
     - p4d.24xlarge
     - p4de.24xlarge
+    - ml.p4d.24xlarge
+    - ml.p4de.24xlarge
     - p5.48xlarge
     - p5e.48xlarge
+    - ml.p5.48xlarge
+    - ml.p5e.48xlarge
     - trn1.32xlarge
     - trn1n.32xlarge
+    - ml.trn1.32xlarge
+    - ml.trn1n.32xlarge
     - vt1.24xlarge
     - hpc6a.48xlarge
     - hpc6id.32xlarge
@@ -143,10 +156,14 @@ resources:
      memory:  20Mi
 nodeSelector: {}
 #  efa: present
-tolerations: []
+tolerations:
 # - key: aws.amazon.com/efa
 #   operator: Exists
 #   effect: NoSchedule
+  - key: sagemaker.amazonaws.com/node-health-status
+    operator: Equal
+    effect: NoSchedule
+    value: Unschedulable
 additionalPodAnnotations: {}
 additionalPodLabels: {}
 nameOverride: ""