from transformers import PretrainedConfig
import torch


class ImpressoConfig(PretrainedConfig):
    model_type = "langident"

    def __init__(
        self,
        filename=None,
        **kwargs,
    ):
        super().__init__(**kwargs)
        self.filename = filename


# Register the configuration with the transformers library
ImpressoConfig.register_for_auto_class()