From f2e49b1d575b3da4367ba4d60b95187f270d42c9 Mon Sep 17 00:00:00 2001
From: Jianqi Pan <jannchie@gmail.com>
Date: Tue, 14 Nov 2023 14:32:05 +0900
Subject: [PATCH] fix: adaptation to older versions of pytroch
---
comfy/sd1_clip.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/comfy/sd1_clip.py b/comfy/sd1_clip.py
index 7db7ee0f..af621b2d 100644
--- 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)