diff --git a/pyproject.toml b/pyproject.toml
index 96fc96cf..c342cacc 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -24,6 +24,9 @@ dev = ["pytest", "pytest-cov"]
 [project.urls]
 repository = "https://github.com/yondonfu/comfystream"
 
+[project.scripts]
+comfystream-build-trt = "comfystream.scripts.build_trt:main"
+
 [tool.comfy]
 PublisherId = "livepeer-comfystream"
 DisplayName = "ComfyStream"