wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-17 00:43:48 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Merge 1ce987cf52 into 9da397ea2f

Browse Source
This commit is contained in:
diodiogod 2025-10-18 19:19:42 +03:00 committed by GitHub
commit 6e9e7fb734
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/ops.py
View File

@ -68,7 +68,10 @@ except:
cast_to = comfy.model_management.cast_to #TODO: remove once no more references
if torch.cuda.is_available() and torch.backends.cudnn.is_available() and PerformanceFeature.AutoTune in args.fast:
torch.backends.cudnn.benchmark = True
import sys
# Skip CUDNN benchmark on Python 3.12 due to VRAM allocation issues with model wrappers
if sys.version_info[:2] != (3, 12):
torch.backends.cudnn.benchmark = True
def cast_to_input(weight, input, non_blocking=False, copy=True):
return comfy.model_management.cast_to(weight, input.dtype, input.device, non_blocking=non_blocking, copy=copy)