Method __init__

models/ideogram4.py:261–263

(self, model)

class Wrapper(nn.Module):
    def __init__(self, model):
        super().__init__()
        self.model = model

    def forward(self, inputs):
        x_chunk, timesteps, context, attn_mask = inputs