diff --git a/main.py b/main.py index eb64726ea..b6f7d6cef 100644 --- a/main.py +++ b/main.py @@ -248,7 +248,17 @@ import hook_breaker_ac10a0 import comfy.memory_management import comfy.model_patcher -if args.enable_dynamic_vram or (enables_dynamic_vram() and comfy.model_management.is_nvidia()): + +def dynamic_vram_supported(): + if comfy.model_management.is_nvidia(): + return True + if comfy.model_management.is_amd(): + if comfy.model_management.rocm_version >= (7, 14): + return True + return False + + +if args.enable_dynamic_vram or (enables_dynamic_vram() and dynamic_vram_supported()): if (not args.enable_dynamic_vram) and (comfy.model_management.torch_version_numeric < (2, 8)): logging.warning("Unsupported Pytorch detected. DynamicVRAM support requires Pytorch version 2.8 or later. Falling back to legacy ModelPatcher. VRAM estimates may be unreliable especially on Windows") else: