src/transformers/generation/logits_process.py/ForcedEOSTokenLogitsProcessor/__init__
class ForcedEOSTokenLogitsProcessor:    def __init__(self, max_length: int, eos_token_id: Union[int, List[int]]):
        self.max_length = max_length
        if isinstance(eos_token_id, int):
            eos_token_id = [eos_token_id]
        self.eos_token_id = eos_token_id