Function num_to_groups

usr/diff/shallow_diffusion_tts.py:34–40

(num, divisor)

def num_to_groups(num, divisor):
    groups = num // divisor
    remainder = num % divisor
    arr = [divisor] * groups
    if remainder > 0:
        arr.append(remainder)
    return arr


class Residual(nn.Module):