Added triton to windows installer.

BuffMcBigHuge · victorges · commit 36ad26bebedd · 2025-10-23T17:32:45.000-07:00
diff --git a/src/streamdiffusion/tools/install-tensorrt.py b/src/streamdiffusion/tools/install-tensorrt.py
@@ -39,6 +39,10 @@ def install(cu: Optional[Literal["11", "12"]] = get_cuda_major()):
         run_pip(
             "install pywin32==306"
         )
+    if platform.system() == 'Windows' and not is_installed("triton"):
+        run_pip(
+            "install triton-windows<3.5"
+        )
 
 
 if __name__ == "__main__":

Original file line number	Diff line number	Diff line change
`@@ -39,6 +39,10 @@ def install(cu: Optional[Literal["11", "12"]] = get_cuda_major()):`
`39`	`39`	`run_pip(`
`40`	`40`	`"install pywin32==306"`
`41`	`41`	`)`
	`42`	`+ if platform.system() == 'Windows' and not is_installed("triton"):`
	`43`	`+ run_pip(`
	`44`	`+ "install triton-windows<3.5"`
	`45`	`+ )`
`42`	`46`
`43`	`47`
`44`	`48`	`if __name__ == "__main__":`