Create cc_projection.py

cc_projection/cc_projection.py

@@ -0,0 +1,13 @@
+import torch
+from diffusers.configuration_utils import ConfigMixin
+from diffusers.models.modeling_utils import ModelMixin
+
+class CCProjection(ModelMixin, ConfigMixin):
+    def __init__(self, in_channel=1028, out_channel=1024):
+        super().__init__()
+        self.in_channel = in_channel
+        self.out_channel = out_channel
+        self.projection = torch.nn.Linear(in_channel, out_channel)
+
+    def forward(self, x):
+        return self.projection(x)