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

1 year ago · ec3d0ab432
parent c962884a5c f2e49b1d57
commit ec3d0ab432
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)