xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-12 21:36:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge 1ce987cf52c84517e06ea43ec0564590a0b83e9e into 9da397ea2f271080406f0c14cf4f0db7221ddf70

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

3
comfy/ops.py
View File

@ -68,6 +68,9 @@ except:
cast_to = comfy.model_management.cast_to #TODO: remove once no more references 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: if torch.cuda.is_available() and torch.backends.cudnn.is_available() and PerformanceFeature.AutoTune in args.fast:
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 torch.backends.cudnn.benchmark = True
def cast_to_input(weight, input, non_blocking=False, copy=True): def cast_to_input(weight, input, non_blocking=False, copy=True):