fix the conversation and run success;

2024-12-05 00:18:56 -08:00
parent 81f58d87cd
commit dde1273804
7 changed files with 33 additions and 769 deletions
@@ -226,8 +226,6 @@ class Sana(nn.Module):
        )
        self.final_layer = T2IFinalLayer(hidden_size, patch_size, self.out_channels)

-        self.initialize_weights()
-
    def forward(self, x, timestep, y, mask=None, data_info=None, **kwargs):
        """
        Forward pass of Sana.