diff --git a/comfy/customzluda/zluda.py b/comfy/customzluda/zluda.py index 5716b4245..8cac667ba 100644 --- a/comfy/customzluda/zluda.py +++ b/comfy/customzluda/zluda.py @@ -431,6 +431,13 @@ try: import triton.language as tl print(" :: Triton core imported successfully") + # Get triton version, if possible, continue if no version string + try: + version = triton.__version__ + print(f" :: Detected Triton version: {version}") + except AttributeError: + print(" :: Detected Triton version: no info") + # This needs to be up here, so it can disable cudnn before anything can even think about using it torch.backends.cudnn.enabled = os.environ.get("TORCH_BACKENDS_CUDNN_ENABLED", "1").strip().lower() not in {"0", "off", "false", "disable", "disabled", "no"} if torch.backends.cudnn.enabled: