Updated model registration

register_model.py

@@ -6,7 +6,7 @@ from sagvit_config import SAGViTConfig
 from sag_vit_model import SAGViTClassifier
 
 # Register the configuration
-CONFIG_MAPPING.register("sagvit", SAGViTConfig)
+CONFIG_MAPPING.register("sag-vit", SAGViTConfig)
 
 # Register the model
 MODEL_MAPPING.register(SAGViTConfig, SAGViTClassifier)

sagvit_config.py

@@ -1,7 +1,7 @@
 from transformers import PretrainedConfig
 
 class SAGViTConfig(PretrainedConfig):
-    model_type = "sagvit"
+    model_type = "sag-vit"
 
     def __init__(self,
                  d_model=64,