(self, model, peft_config: PeftConfig, adapter_name="default")
| 474 | """ |
| 475 | |
| 476 | def __init__(self, model, peft_config: PeftConfig, adapter_name="default"): |
| 477 | super().__init__(model, peft_config, adapter_name) |
| 478 | if self.modules_to_save is None: |
| 479 | self.modules_to_save = {"classifier", "score"} |
| 480 | else: |
| 481 | self.modules_to_save.update({"classifier", "score"}) |
| 482 | |
| 483 | for name, _ in self.base_model.named_children(): |
| 484 | if any(module_name in name for module_name in self.modules_to_save): |
| 485 | self.cls_layer_name = name |
| 486 | break |
| 487 | |
| 488 | # to make sure classifier layer is trainable |
| 489 | _set_trainable(self, adapter_name) |
| 490 | |
| 491 | def forward( |
| 492 | self, |
nothing calls this directly
no test coverage detected