Upload model

prot_albert_model.py

@@ -28,6 +28,7 @@ class ProtAlBertModel(PreTrainedModel):
         print(f"hidden_size: {hidden_size}")
         print(f"num_labels: {num_labels}")
         print(f"CONFIG HIDDEN : {self.config.hidden_size}")
+        num_labels = 10 if type(num_labels) is not int else num_labels
         self.last_layer = nn.Sequential(
             nn.Dropout(0.1),
             nn.LayerNorm(self.config.hidden_size),