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"]
|