Function swap_scale_shift

scripts/convert_aura_flow_to_diffusers.py:25–28

(weight, dim)

# similar to SD3 but only for the last norm layer
def swap_scale_shift(weight, dim):
    shift, scale = weight.chunk(2, dim=0)
    new_weight = torch.cat([scale, shift], dim=0)
    return new_weight


def convert_transformer(state_dict):