def __init__(self, model: torch.nn.Module):
                        """
                        Initialization by storing the input model.

                        :param model: PyTorch model. The forward function of the model must return the logit output.
                        """
                        super().__init__()
                        self._model = model

                        Initialization by storing the input model.

                        :param model: PyTorch model. The forward function of the model must return the logit output.
                        
art/estimators/classification/hugging_face.py