Function modulate

src/models/transformer_3d.py:27–28

(x, shift, scale)

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

class FinalLayer(nn.Module):
    def __init__(self, hidden_size):