Skip to content

Commit 2c2aa40

Browse files
Log message for cudnn disable on AMD. (Comfy-Org#10418)
1 parent a4787ac commit 2c2aa40

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

comfy/model_management.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -333,6 +333,7 @@ def amd_min_version(device=None, min_rdna_version=0):
333333
try:
334334
if is_amd():
335335
torch.backends.cudnn.enabled = False # Seems to improve things a lot on AMD
336+
logging.info("Set: torch.backends.cudnn.enabled = False for better AMD performance.")
336337
try:
337338
rocm_version = tuple(map(int, str(torch.version.hip).split(".")[:2]))
338339
except:

0 commit comments

Comments
 (0)