Add config param to model

model.py

@@ -18,7 +18,8 @@ class MultiLabelAttention(torch.nn.Module):
 class BertMesh(PreTrainedModel):
     def __init__(
         self,
-        pretrained_model,
+        config,
+        pretrained_model="microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract",
         num_labels=28761,
         hidden_size=512,
         dropout=0,