From 75ec851324f5e171fb93b36371e0a4bbbc6fa202 Mon Sep 17 00:00:00 2001 From: "Chowdhury, Hisham" Date: Thu, 6 Feb 2025 12:23:33 -0800 Subject: [PATCH] fix formating --- comfy/clip_model.py | 2 +- comfy/model_management.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/comfy/clip_model.py b/comfy/clip_model.py index d176b3c7a..466d5887c 100644 --- a/comfy/clip_model.py +++ b/comfy/clip_model.py @@ -108,7 +108,7 @@ class CLIPTextModel_(torch.nn.Module): causal_mask = torch.empty(x.shape[1], x.shape[1], dtype=x.dtype, device=x.device).triu_(1) else: causal_mask = torch.empty(x.shape[1], x.shape[1], dtype=x.dtype, device=x.device).fill_(float("-inf")).triu_(1) - + if mask is not None: mask += causal_mask else: diff --git a/comfy/model_management.py b/comfy/model_management.py index 95fd39f6a..d571e7bff 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -997,7 +997,7 @@ def is_directml_enabled(): global directml_enabled if directml_enabled: return True - + return False def should_use_fp16(device=None, model_params=0, prioritize_performance=True, manual_cast=False):