def METHOD_NAME(self, module: Module) -> None:
    weight = self.compute_weight(module)
    delattr(module, self.name)
    del module._parameters[self.name + "_g"]
    del module._parameters[self.name + "_v"]
    setattr(module, self.name, flow.nn.Parameter(weight))