Method __init__

rl_engine/alignment/model_wrappers.py:80–82

(self, model: torch.nn.Module)

    """Standard adapter for the trainable policy model used by RL losses."""

    def __init__(self, model: torch.nn.Module):
        super().__init__()
        self.model = model

    def forward(self, input_ids: torch.Tensor, **model_kwargs: Any) -> Any:
        return self.model(input_ids, **model_kwargs)