dc8ca3d 2bf5d65 919b526 2bf5d65
1
2
3
4
5
6
7
8
9
# __init__.py from transformers import AutoConfig, AutoModel from .custom_model import CustomConfig, CustomModel # Register the custom classes AutoConfig.register("custom_model", CustomConfig) AutoModel.register(CustomConfig, CustomModel)
