Update toonmage/fluxpipeline.py

toonmage/fluxpipeline.py

@@ -34,11 +34,11 @@ class ToonMagePipeline(nn.Module):
             num_ca += 1
         if 38 % single_interval != 0:
             num_ca += 1
-        self.toonmage_ca = nn.ModuleList([
+        self.pulid_ca = nn.ModuleList([
             PerceiverAttentionCA().to(self.device, self.weight_dtype) for _ in range(num_ca)
         ])
 
-        dit.toonmage_ca = self.toonmage_ca
+        dit.pulid_ca = self.pulid_ca
         dit.toonmage_double_interval = double_interval
         dit.toonmage_single_interval = single_interval