Fix last pr.

comfyanonymous · Nov 14, 2023 · 728613b · 728613b
1 parent ec3d0ab
commit 728613b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/comfy/sd1_clip.py b/comfy/sd1_clip.py
@@ -173,7 +173,7 @@ def forward(self, tokens):
         if getattr(self.transformer, self.inner_name).final_layer_norm.weight.dtype != torch.float32:
             precision_scope = torch.autocast
         else:
-            precision_scope = lambda a, b: contextlib.nullcontext(a)
+            precision_scope = lambda a, dtype: contextlib.nullcontext(a)
 
         with precision_scope(model_management.get_autocast_device(device), dtype=torch.float32):
             attention_mask = None