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

ec3d0ab4 · comfyanonymous · c962884a · f2e49b1d · ec3d0ab4
Commit ec3d0ab4 authored Nov 14, 2023 by comfyanonymous
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

comfy/sd1_clip.py comfy/sd1_clip.py +1 -1

No files found.
--- 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)