Method __init__

sat/sgm/modules/diffusionmodules/openaimodel.py:486–488

(self, dim)

class Timestep(nn.Module):
    def __init__(self, dim):
        super().__init__()
        self.dim = dim

    def forward(self, t):
        return timestep_embedding(t, self.dim)