Method requires_grad_

networks/lora_fa.py:97–100

(self, requires_grad: bool = True)

        return trainable_params

    def requires_grad_(self, requires_grad: bool = True):
        self.lora_up.requires_grad_(requires_grad)
        self.lora_down.requires_grad_(False)
        return self

    def apply_to(self):
        self.org_forward = self.org_module.forward