Browse Source

fix: adaptation to older versions of pytroch

pull/1964/head
Jianqi Pan 1 year ago
parent
commit
f2e49b1d57
  1. 2
      comfy/sd1_clip.py

2
comfy/sd1_clip.py

@ -175,7 +175,7 @@ class SDClipModel(torch.nn.Module, ClipTokenWeightEncoder):
else:
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
if self.enable_attention_masks:
attention_mask = torch.zeros_like(tokens)

Loading…
Cancel
Save