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):