File size: 314 Bytes
44d9cf1
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
# friday/__init__.py
from .model.friday_arch import FridayConfig, FridayForCausalLM

from transformers import AutoConfig, AutoModelForCausalLM
AutoConfig.register(FridayConfig.model_type, FridayConfig)
AutoModelForCausalLM.register(FridayConfig, FridayForCausalLM)

__all__ = ["FridayConfig", "FridayForCausalLM"]