Fix missing variable.

2023-02-25 12:19:03 -05:00 · 2023-02-25 12:19:03 -05:00 · d2da346b0b
parent 1144ed5792
commit d2da346b0b
1 changed files with 1 additions and 0 deletions
--- a/comfy/sd.py
+++ b/comfy/sd.py
@ -535,6 +535,7 @@ class T2IAdapter:
        output_dtype = x_noisy.dtype
        out = {'input':[]}

+        autocast_enabled = torch.is_autocast_enabled()
        for i in range(len(self.control_input)):
            key = 'input'
            x = self.control_input[i] * self.strength