File size: 377 Bytes
ae8276c f6eb553 ae8276c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
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()
|