File size: 295 Bytes
31a17dc 32db49c 31a17dc 32db49c | 1 2 3 4 5 6 7 8 9 | from transformers import AutoConfig, AutoModel
from modeling_sagvit import SAGViTConfig, SAGViTClassifier
# Register Custom Model and Config
print("Registering model")
AutoConfig.register("sagvit", SAGViTConfig)
AutoModel.register(SAGViTConfig, SAGViTClassifier)
print("Registration complete")
|