File size: 223 Bytes
b3d4005
 
 
 
4c31efb
b3d4005
 
 
1
2
3
4
5
6
7
8
9
from transformers import PretrainedConfig
import torch

class ImpressoConfig(PretrainedConfig):
    model_type = "floret"

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