From cba1b4fcaaf3569f7464ed704f8196c02a040978 Mon Sep 17 00:00:00 2001
From: Cursor Agent <cursoragent@cursor.com>
Date: Sun, 29 Jun 2025 00:21:53 +0000
Subject: [PATCH] Add CLI script for building TensorRT engine in pyproject.toml

Co-authored-by: john <john@eliteencoder.net>
---
 pyproject.toml | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pyproject.toml b/pyproject.toml
index 96fc96cf1..c342cacc2 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"