(
self,
ref_model: Union[PreTrainedModel, torch.nn.Module],
accelerator: Optional[Accelerator],
)
| 96 | """ |
| 97 | |
| 98 | def __init__( |
| 99 | self, |
| 100 | ref_model: Union[PreTrainedModel, torch.nn.Module], |
| 101 | accelerator: Optional[Accelerator], |
| 102 | ): |
| 103 | self.accelerator = accelerator |
| 104 | self.ref_model = ref_model |
| 105 | |
| 106 | @staticmethod |
| 107 | def _sync_target_model(model, target_model, alpha): |
nothing calls this directly
no outgoing calls
no test coverage detected