def __init__()

in src/optimum/nvidia/pipelines/text_generation.py [0:0]

• 6 lines of code
• 3 McCabe index (conditional complexity)

    def __init__(self, model: CausalLM, tokenizer: PreTrainedTokenizer):
        super().__init__()

        if tokenizer.eos_token and not tokenizer.pad_token:
            tokenizer.pad_token = tokenizer.eos_token

        self.tokenizer = tokenizer
        self._runtime = model