File size: 438 Bytes
0094a2a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
# -*- coding: utf-8 -*-
from transformers import AutoConfig, AutoModel, AutoModelForCausalLM
from fla.models.scan.configuration_scan import SCANConfig
from fla.models.scan.modeling_scan import SCANForCausalLM, SCANModel
AutoConfig.register(SCANConfig.model_type, SCANConfig)
AutoModel.register(SCANConfig, SCANModel)
AutoModelForCausalLM.register(SCANConfig, SCANForCausalLM)
__all__ = ['SCANConfig', 'SCANForCausalLM', 'SCANModel']
|