diff --git a/comfy/model_management.py b/comfy/model_management.py index 032041812..e134440f5 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -237,10 +237,10 @@ try: except: pass -if ENABLE_PYTORCH_ATTENTION: - torch.backends.cuda.enable_math_sdp(True) - torch.backends.cuda.enable_flash_sdp(True) - torch.backends.cuda.enable_mem_efficient_sdp(True) +#if ENABLE_PYTORCH_ATTENTION: +# torch.backends.cuda.enable_math_sdp(True) +# torch.backends.cuda.enable_flash_sdp(True) +# torch.backends.cuda.enable_mem_efficient_sdp(True) try: if is_nvidia() and args.fast: