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