src/transformers/generation/flax_logits_process.py/FlaxForcedEOSTokenLogitsProcessor/__init__
class FlaxForcedEOSTokenLogitsProcessor:    def __init__(self, max_length: int, eos_token_id: int):
        self.max_length = max_length
        self.eos_token_id = eos_token_id