(self, model, peft_config: PeftConfig = None, adapter_name="default")
| 1046 | """ |
| 1047 | |
| 1048 | def __init__(self, model, peft_config: PeftConfig = None, adapter_name="default"): |
| 1049 | super().__init__(model, peft_config, adapter_name) |
| 1050 | if self.modules_to_save is None: |
| 1051 | self.modules_to_save = {"classifier", "score"} |
| 1052 | else: |
| 1053 | self.modules_to_save.update({"classifier", "score"}) |
| 1054 | |
| 1055 | for name, _ in self.base_model.named_children(): |
| 1056 | if any(module_name in name for module_name in self.modules_to_save): |
| 1057 | self.cls_layer_name = name |
| 1058 | break |
| 1059 | |
| 1060 | # to make sure classifier layer is trainable |
| 1061 | _set_trainable(self, adapter_name) |
| 1062 | |
| 1063 | def forward( |
| 1064 | self, |
nothing calls this directly
no test coverage detected