diff --git a/pdm.lock b/pdm.lock
index 923198d9..f82b49c3 100644
--- a/pdm.lock
+++ b/pdm.lock
@@ -5,7 +5,7 @@
 groups = ["default", "docs", "lint", "test"]
 strategy = ["cross_platform", "inherit_metadata"]
 lock_version = "4.4.1"
-content_hash = "sha256:4656b8dfc9408c9b8a5713869b5d622a4446b296cda7bbf319c08d2b4945c7c7"
+content_hash = "sha256:a8125f72370ebdf9f5dc8b24aa1dbe72dad9dc31f4346be953476ebc9de23df6"
 
 [[package]]
 name = "aiohttp"
@@ -1666,9 +1666,6 @@ files = [
 name = "llama-index"
 version = "0.9.28"
 requires_python = ">=3.8.1,<4.0"
-git = "https://github.com/Zipstack/llama-index.git"
-ref = "v0.9.28-unstract.1"
-revision = "ce8b60d096463cf629460e8d6472e8bbcfe91104"
 summary = "Interface between LLMs and your data"
 groups = ["default"]
 dependencies = [
@@ -1687,10 +1684,15 @@ dependencies = [
     "pandas",
     "requests>=2.31.0",
     "tenacity<9.0.0,>=8.2.0",
+    "tensorrt-llm<0.8.0,>=0.7.1; python_version == \"3.10\"",
     "tiktoken>=0.3.3",
     "typing-extensions>=4.5.0",
     "typing-inspect>=0.8.0",
 ]
+files = [
+    {file = "llama_index-0.9.28-py3-none-any.whl", hash = "sha256:fe68d102e9796eac54e7b6e4bc470fa63e0f4b8df6570f9ca7e0b2a644ffd49a"},
+    {file = "llama_index-0.9.28.tar.gz", hash = "sha256:32504aac009c9c5addee36a5368cf4c308a898738bb2135bfcd032874f4dba26"},
+]
 
 [[package]]
 name = "log-symbols"
@@ -3456,6 +3458,17 @@ files = [
     {file = "tenacity-8.2.3.tar.gz", hash = "sha256:5398ef0d78e63f40007c1fb4c0bff96e1911394d2fa8d194f77619c05ff6cc8a"},
 ]
 
+[[package]]
+name = "tensorrt-llm"
+version = "0.7.1"
+requires_python = ">=3.7, <4"
+summary = ""
+groups = ["default"]
+marker = "python_version == \"3.10\""
+files = [
+    {file = "tensorrt-llm-0.7.1.tar.gz", hash = "sha256:10275e2585484f138b43576acddba42ebee073fbb1665fe0954032e421e26e08"},
+]
+
 [[package]]
 name = "termcolor"
 version = "2.4.0"
diff --git a/pyproject.toml b/pyproject.toml
index 9590f6e8..874a15f0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -16,7 +16,7 @@ dependencies = [
     "python-dotenv==1.0.0",
     # LLM Triad
     "unstract-adapters~=0.2.0",
-    "llama-index @ git+https://github.com/Zipstack/llama-index.git@v0.9.28-unstract.1",
+    "llama-index==0.9.28",
     "tiktoken~=0.4.0",
     "transformers==4.37.0",
     # LLM Whisperer dependencies