Skip to content

GitLab

"git@developer.sourcefind.cn:dadigang/Ventoy.git" did not exist on "32a964ededf704cd374dbc96681233a8bf8b6eb4"
Commit ec3d0ab4 authored by comfyanonymous's avatar comfyanonymous
Browse files

Merge branch 'master' of https://github.com/Jannchie/ComfyUI

parents c962884a f2e49b1d
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
comfy/sd1_clip.py
View file @ ec3d0ab4
...@@ -175,7 +175,7 @@ class SDClipModel(torch.nn.Module, ClipTokenWeightEncoder): ...@@ -175,7 +175,7 @@ class SDClipModel(torch.nn.Module, ClipTokenWeightEncoder):
else: else:
precision_scope = lambda a, b: contextlib.nullcontext(a) precision_scope = lambda a, b: contextlib.nullcontext(a)
with precision_scope(model_management.get_autocast_device(device), torch.float32): with precision_scope(model_management.get_autocast_device(device), dtype=torch.float32):
attention_mask = None attention_mask = None
if self.enable_attention_masks: if self.enable_attention_masks:
attention_mask = torch.zeros_like(tokens) attention_mask = torch.zeros_like(tokens)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment