diff --git a/meta-digi-arm/conf/layer.conf b/meta-digi-arm/conf/layer.conf
index 9b363fb63..d34c478ae 100644
--- a/meta-digi-arm/conf/layer.conf
+++ b/meta-digi-arm/conf/layer.conf
@@ -36,6 +36,13 @@ BBMASK += " \
     meta-st-stm32mp/recipes-core/systemd/systemd-conf_%.bbappend \
 "
 
+# Tweak stm32mpu-ai dependencies to use ST's machine learning packages
+LAYERDEPENDS_stm32mpu-ai:remove = "st-openstlinux"
+BBMASK += " \
+    meta-st-stm32mpu-ai/recipes-st/images/st-image-ai-userfs.bb \
+    meta-st-stm32mpu-ai/recipes-st/images/st-image-ai.bb \
+"
+
 # Digi's General and Open Source license agreements
 DIGI_EULA_FILE = "${LAYERDIR}/DIGI_EULA"
 DIGI_OPEN_EULA_FILE = "${LAYERDIR}/DIGI_OPEN_EULA"