From 57b06d1e64428f418718bb524f77377e932093c1 Mon Sep 17 00:00:00 2001
From: Patrick Ge <53650488+cococo2000@users.noreply.github.com>
Date: Fri, 19 Jul 2024 22:19:35 +0800
Subject: [PATCH] Update install-tensorrt.py: fix onnx-graphsurgeon==0.3.26 Not
 Found

Updated onnx-graphsurgeon version in the install function from 0.3.26 to 0.5.2
---
 src/streamdiffusion/tools/install-tensorrt.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/streamdiffusion/tools/install-tensorrt.py b/src/streamdiffusion/tools/install-tensorrt.py
index 182871c4..8916baf3 100644
--- a/src/streamdiffusion/tools/install-tensorrt.py
+++ b/src/streamdiffusion/tools/install-tensorrt.py
@@ -40,7 +40,7 @@ def install(cu: Optional[Literal["11", "12"]] = get_cuda_version_from_torch()):
         )
     if not is_installed("onnx_graphsurgeon"):
         run_pip(
-            "install onnx-graphsurgeon==0.3.26 --extra-index-url https://pypi.ngc.nvidia.com"
+            "install onnx-graphsurgeon==0.5.2 --extra-index-url https://pypi.ngc.nvidia.com"
         )
     if platform.system() == 'Windows' and not is_installed("pywin32"):
         run_pip(