fix: adaptation to older versions of pytroch

1 year ago · f2e49b1d57
parent 8509bd58b4
commit f2e49b1d57
1 changed files with 1 additions and 1 deletions
--- a/comfy/sd1_clip.py
+++ b/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)