From dbc2a4ba29a4df0cd61e1a88d9c27b0473b47d3c Mon Sep 17 00:00:00 2001 From: doctorpangloss <@hiddenswitch.com> Date: Fri, 28 Jun 2024 18:45:15 -0700 Subject: [PATCH] Disable this lint warning --- comfy/model_management.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/model_management.py b/comfy/model_management.py index 428db65ff..a1f078391 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -987,7 +987,7 @@ def soft_empty_cache(force=False): if cpu_state == CPUState.MPS: torch.mps.empty_cache() elif is_intel_xpu(): - torch.xpu.empty_cache() + torch.xpu.empty_cache() # pylint: disable=no-member elif torch.cuda.is_available(): if force or is_nvidia(): # This seems to make things worse on ROCm so I only do it for cuda torch.cuda.empty_cache()