Upload model

prot_albert_model.py

@@ -5,11 +5,13 @@ from transformers import AlbertModel, AutoConfig
 from transformers import PreTrainedModel, AutoModel, AutoConfig
 
 # from protalbert_pfam.prot_albert_config import ProtAlBertConfig
-from protalbert_pfam.prot_albert_config import ProtAlBertConfig
+#from protalbert_pfam.prot_albert_config import ProtAlBertConfig
+from transformers.models.albert.configuration_albert import AlbertConfig
+
 
 
 class ProtAlBertModel(PreTrainedModel):
-    config_class = ProtAlBertConfig
+    config_class = AlbertConfig
     def __init__(self, num_labels: int, hidden_size: int, model_name: str, *args, **kwargs):
         """
         Initialise the model.