implemented t2iadapter weights functionality

2023-08-01 10:01:54 -05:00
parent 935ef4a517
commit c15865e0a6
1 changed files with 2 additions and 2 deletions
@@ -248,10 +248,10 @@ class T2IAdapterAdvanced(ControlBase):
        out = {'input':[]}

        autocast_enabled = torch.is_autocast_enabled()
-        print(f"$$$$ t2i control_input len: {len(self.control_input)}")
+        #print(f"$$$$ t2i control_input len: {len(self.control_input)}")
        for i in range(len(self.control_input)):
            key = 'input'
-            x = self.control_input[i] * self.strength
+            x = self.control_input[i] * self.strength * self.weights[i]
            if x.dtype != output_dtype and not autocast_enabled:
                x = x.to(output_dtype)