298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-07-09 00:37:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix formating

Browse Source
This commit is contained in:
Chowdhury, Hisham 2025-02-06 12:23:33 -08:00
parent b4c7540fbc
commit 75ec851324
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/clip_model.py
View File

@ -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) causal_mask = torch.empty(x.shape[1], x.shape[1], dtype=x.dtype, device=x.device).triu_(1)
else: else:
causal_mask = torch.empty(x.shape[1], x.shape[1], dtype=x.dtype, device=x.device).fill_(float("-inf")).triu_(1) 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: if mask is not None:
mask += causal_mask mask += causal_mask
else: else:

2
comfy/model_management.py
View File

@ -997,7 +997,7 @@ def is_directml_enabled():
global directml_enabled global directml_enabled
if directml_enabled: if directml_enabled:
return True return True
return False return False
def should_use_fp16(device=None, model_params=0, prioritize_performance=True, manual_cast=False): def should_use_fp16(device=None, model_params=0, prioritize_performance=True, manual_cast=False):