diff --git a/.gitmodules b/.gitmodules
index b85ca99e1a54fc4460daa421913f1fe0416230fa..01625342abc79bfced64ceba3992dee31089493c 100644
--- a/.gitmodules
+++ b/.gitmodules
@@ -1,4 +1,4 @@
 [submodule "tests/mindformers"]
 	path = tests/mindformers
 	url = https://gitee.com/mindspore/mindformers.git
-	branch = br_infer_boom
+	branch = br_feature_infer
diff --git a/tests/mindformers b/tests/mindformers
index a96a0cf9a0b1e3efee819c17dcf050604caa3512..0789b056aa5ef986f0300a3139713aaf63631e41 160000
--- a/tests/mindformers
+++ b/tests/mindformers
@@ -1 +1 @@
-Subproject commit a96a0cf9a0b1e3efee819c17dcf050604caa3512
+Subproject commit 0789b056aa5ef986f0300a3139713aaf63631e41