Method __init__

diffsynth/models/sd_unet.py:7–9

(self, num_channels)

class Timesteps(torch.nn.Module):
    def __init__(self, num_channels):
        super().__init__()
        self.num_channels = num_channels

    def forward(self, timesteps):
        half_dim = self.num_channels // 2