Skip to content

fix: skip PatchTritonVAE registration when triton is not installed - #727

Open
hcwhan wants to merge 1 commit into
kijai:mainfrom
hcwhan:fix/skip-patch-triton-vae-when-triton-missing
Open

fix: skip PatchTritonVAE registration when triton is not installed#727
hcwhan wants to merge 1 commit into
kijai:mainfrom
hcwhan:fix/skip-patch-triton-vae-when-triton-missing

Conversation

@hcwhan

@hcwhan hcwhan commented Aug 7, 2026

Copy link
Copy Markdown

Summary

  • Check whether the triton package is available before importing PatchTritonVAE
  • Avoid startup warning tracebacks on non-CUDA environments (e.g. AMD ROCm) where Triton is not installed
  • When Triton is missing, log an info message and skip the node registration instead of attempting import

Motivation

PatchTritonVAE depends on Triton, which is typically unavailable outside NVIDIA CUDA setups. The current unconditional import attempt triggers a warning with full traceback on every startup, even though the node cannot be used in those environments.

Test plan

  • Start ComfyUI without Triton installed: no warning traceback; info log about skipping PatchTritonVAE
  • Start ComfyUI with Triton installed: PatchTritonVAE node still registers normally
  • Other KJNodes nodes continue to load as before

Check for triton availability before importing triton_vae so non-CUDA
environments (e.g. AMD ROCm) avoid a noisy warning traceback on startup.
When triton is missing, log an info message and skip the node instead.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant