| from transformers import PretrainedConfig | |
| import torch | |
| class ImpressoConfig(PretrainedConfig): | |
| model_type = "lang_detect" | |
| def __init__( | |
| self, | |
| filename=None, | |
| **kwargs, | |
| ): | |
| super().__init__(**kwargs) | |
| self.filename = filename | |
| # Register the configuration with the transformers library | |
| ImpressoConfig.register_for_auto_class() | |