wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-23 18:13:28 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

run triton import with version check when enable_triton_backend launch arg is used to check if it actually works.

Browse Source
This commit is contained in:
silveroxides 2026-03-05 11:44:52 +01:00
parent 7506c7baed
commit 2329f8c925
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/quant_ops.py
View File

@ -26,7 +26,7 @@ try:
if args.enable_triton_backend: if args.enable_triton_backend:
try: try:
import triton import triton
logging.info(f"Found triton package installed. Enabling comfy-kitchen triton backend.") logging.info("Found triton %s. Enabling comfy-kitchen triton backend.", triton.__version__)
except ImportError as e: except ImportError as e:
logging.error(f"Failed to import triton, Error: {e}, the comfy-kitchen triton backend will not be available.") logging.error(f"Failed to import triton, Error: {e}, the comfy-kitchen triton backend will not be available.")
ck.registry.disable("triton") ck.registry.disable("triton")