diff --git a/.gitmodules b/.gitmodules
index 70ad46f..fa859d2 100644
--- a/.gitmodules
+++ b/.gitmodules
@@ -1,3 +1,4 @@
 [submodule "tensorrt_llm"]
 	path = tensorrt_llm
 	url = https://github.com/NVIDIA/TensorRT-LLM.git
+	branch = main