Method __init__

sat/sgm/modules/diffusionmodules/openaimodel.py:1225–1229

(self, in_channels=3, model_channels=64)

    class Dummy(nn.Module):
        def __init__(self, in_channels=3, model_channels=64):
            super().__init__()
            self.input_blocks = nn.ModuleList(
                [TimestepEmbedSequential(conv_nd(2, in_channels, model_channels, 3, padding=1))]
            )

    model = UNetModel(
        use_checkpoint=True,