from transformers import GenerationConfig class MyGenerationConfig(GenerationConfig): def __init__(self, **kwargs): super().__init__(**kwargs) # Set the non-default generation parameters self.early_stopping = True self.forced_eos_token_id = 2