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

Fix torch.compiler.is_compiling missing on torch 2.3 and earlier

Browse Source
This commit is contained in:
doctorpangloss 2025-10-22 13:37:20 -07:00
parent 6810282be4
commit 6954e3e247
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
comfy/ops.py
View File

@ -29,10 +29,13 @@ from .float import stochastic_rounding
logger = logging.getLogger(__name__)
_RUN_EVERY_OP_ENABLED = model_management.torch_version_numeric >= (2, 5)
def run_every_op():
# this is available on torch 2.3
if torch.compiler.is_compiling(): # pylint: disable=no-member
global _RUN_EVERY_OP_ENABLED
# this is not available on torch 2.3 by testing
if not _RUN_EVERY_OP_ENABLED or torch.compiler.is_compiling(): # pylint: disable=no-member
return
throw_exception_if_processing_interrupted()