Function modulate

src/models/attention.py:18–19

(x, shift, scale)

    xformers = None

def modulate(x, shift, scale):
    return x * (1 + scale.unsqueeze(1)) + shift.unsqueeze(1)

class TemporalBasicTransformerBlock(nn.Module):
    def __init__(