Open1

"line 626, in _unsqueeze_helper if _is_constant(axes_i[0]): IndexError: list index out of range"

PINTOPINTO
# self.norm = nn.GroupNorm(1, self.latent_dim)
self.norm = nn.LayerNorm(self.latent_dim)