wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-12-20 03:23:00 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Add env TORCH_AMD_CUDNN_ENABLED (fix typo)

Browse Source
This commit is contained in:
Christopher Anderson 2025-11-08 20:45:32 +11:00
parent 90ba86a95f
commit de408d5f2c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_management.py
View File

@ -343,7 +343,7 @@ try:
"0", "off", "false", "disable", "disabled", "no"} "0", "off", "false", "disable", "disabled", "no"}
if not torch.backends.cudnn.enabled: if not torch.backends.cudnn.enabled:
logging.info( logging.info(
"ComfyUI has set torch.backends.cudnn.enabled to False for better AMD performance. Set environment var TORCH_AMD_CUDDNN_ENABLED=1 to enable it again.") "ComfyUI has set torch.backends.cudnn.enabled to False for better AMD performance. Set environment var TORCH_AMD_CUDNN_ENABLED=1 to enable it again.")
try: try:
rocm_version = tuple(map(int, str(torch.version.hip).split(".")[:2])) rocm_version = tuple(map(int, str(torch.version.hip).split(".")[:2]))