update things

.github/workflows/build.yaml

@@ -56,6 +56,14 @@ jobs:
         run: |
           echo "Version: $VERSION"
 
+      - name: Git Clone TensorRT
+        run: git clone https://github.com/NVIDIA/TensorRT
+
+      - name: CD To Polygraphy and Install Latest Polygraphy
+        run: |
+          cd TensorRT/tools/Polygraphy
+          ./install.ps1
+
       - name: Create Release
         id: create_release
         uses: actions/create-release@v1

build.py

@@ -27,6 +27,13 @@ def install_requirements():
         check=True,
     )
 
+def install_polygraphy():
+    print("Installing Polygraphy...")
+    cwd = os.getcwd()
+    os.chdir("TensorRT/Tools/Polygraphy")
+    subprocess.run(["powershell", "-File", "install.ps1"], check=True)
+    os.chdir(cwd)
+
 
 def install_pyinstaller():
     print("Installing PyInstaller...")

requirements.txt

@@ -12,6 +12,7 @@ spandrel
 yt-dlp
 requests
 tqdm
+tensorrt
 colorama
 einops
 scikit-image