Upload model

config.json

@@ -2,6 +2,10 @@
   "architectures": [
     "VnSmartphoneAbsaModel"
   ],
+  "auto_map": {
+    "AutoConfig": "configuration_vnsabsa.VnSmartphoneAbsaConfig",
+    "AutoModel": "modeling_vnsabsa.VnSmartphoneAbsaModel"
+  },
   "embed_dim": 768,
   "encoder_dropout": 0.1,
   "fc_dropout": 0.4,

configuration_vnsabsa.py

@@ -1,6 +1,6 @@
 from transformers import PretrainedConfig
 
-class VnSmartphoneBERTConfig(PretrainedConfig):
+class VnSmartphoneAbsaConfig(PretrainedConfig):
     model_type = "vnsabsa"
     def __init__(
         self, 

modeling_vnsabsa.py

@@ -2,17 +2,17 @@ from transformers import PreTrainedModel
 from modules import SmartphoneBERT
 import torch
 
-from .configuration_vnsabsa import VnSmartphoneBERTConfig
+from .configuration_vnsabsa import VnSmartphoneAbsaConfig
 
 from typing import Tuple
 
 
-class VnSmartphoneBERTModel(PreTrainedModel):
-    config_class = VnSmartphoneBERTConfig
+class VnSmartphoneAbsaModel(PreTrainedModel):
+    config_class = VnSmartphoneAbsaConfig
     
     def __init__(
         self,
-        config: VnSmartphoneBERTConfig
+        config: VnSmartphoneAbsaConfig
     ):
         super().__init__(config)
         self.model = SmartphoneBERT(