Tensor size error

[arnicas]

hi, this is amazing..... but in the colab as soon as multiconditioning starts, it throws this error:

/content/Paella/modules.py in forward(self, x, c, r)
145 r_embed = self.gen_r_embedding(r)
146 x = self.embedding(x).permute(0, 3, 1, 2)
--> 147 s = torch.cat([c, r_embed], dim=-1)[:, :, None, None]
148 level_outputs = self.down_encode(x, s)
149 x = self._up_decode(level_outputs, s)

RuntimeError: Tensors must have same number of dimensions: got 4 and 2

[dome272]

Hey there,
sorry I forgot to push the last changes today. It should work now!