Method forward

rl_engine/alignment/model_wrappers.py:84–85

(self, input_ids: torch.Tensor, **model_kwargs: Any)

        self.model = model

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

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