fix-copies

Author: Matrix_K

src/diffusers/models/controlnets/controlnet_z_image.py

@@ -329,7 +329,7 @@ def __call__(self, ids: torch.Tensor):
         result = []
         for i in range(len(self.axes_dims)):
             index = ids[:, i]
-            result.append(self.freqs_cis[i][index])
+            result.append(torch.complex(self.freqs_cis[i].real[index], self.freqs_cis[i].imag[index]))
         return torch.cat(result, dim=-1)