diff --git a/comfy/model_management.py b/comfy/model_management.py index 7467391cd9bf..a2c318ec3e00 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -333,6 +333,7 @@ def amd_min_version(device=None, min_rdna_version=0): try: if is_amd(): torch.backends.cudnn.enabled = False # Seems to improve things a lot on AMD + logging.info("Set: torch.backends.cudnn.enabled = False for better AMD performance.") try: rocm_version = tuple(map(int, str(torch.version.hip).split(".")[:2])) except: