Upload 4 files

config.json

@@ -1,21 +1,21 @@
-{
-  "architectures": [
-    "PDeepPPModel"
-  ],
-  "auto_map": {
-    "AutoConfig": "configuration_PDeepPP.PDeepPPConfig",
-    "AutoModel": "modeling_PDeepPP.PDeepPPModel"
-  },
-  "dropout": 0.3,
-  "esm_ratio": 1.0,
-  "hidden_size": 256,
-  "input_size": 1280,
-  "lambda_": 0.96,
-  "model_type": "PDeepPP",
-  "num_heads": 8,
-  "num_transformer_layers": 4,
-  "output_size": 128,
-  "ptm_type": "ACE",
-  "torch_dtype": "float32",
-  "transformers_version": "4.35.2"
-}
+{
+  "architectures": [
+    "PDeepPPModel"
+  ],
+  "auto_map": {
+    "AutoConfig": "configuration_pdeeppp.PDeepPPConfig",
+    "AutoModel": "modeling_PDeepPP.PDeepPPModel"
+  },
+  "dropout": 0.3,
+  "esm_ratio": 1.0,
+  "hidden_size": 256,
+  "input_size": 1280,
+  "lambda_": 0.96,
+  "model_type": "PDeepPP",
+  "num_heads": 8,
+  "num_transformer_layers": 4,
+  "output_size": 128,
+  "ptm_type": "ACE",
+  "torch_dtype": "float32",
+  "transformers_version": "4.35.2"
+}

configuration_pdeeppp.py

@@ -0,0 +1,37 @@
+import os
+from typing import Dict, List, Optional, Union
+
+from transformers.configuration_utils import PretrainedConfig
+from transformers.utils import logging
+
+logger = logging.get_logger(__name__)
+
+class PDeepPPConfig(PretrainedConfig):
+
+    model_type = "PDeepPP"
+
+    def __init__(
+        self,
+        input_size=1280,
+        output_size=128,
+        num_heads=8,
+        hidden_size=256,
+        num_transformer_layers=4,
+        dropout=0.3,
+        ptm_type="ACE",
+        esm_ratio=0.96,
+        lambda_=1,
+        **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.input_size = input_size
+        self.output_size = output_size
+        self.num_heads = num_heads
+        self.hidden_size = hidden_size
+        self.num_transformer_layers = num_transformer_layers
+        self.dropout = dropout
+        self.ptm_type = ptm_type
+        self.esm_ratio = esm_ratio
+        self.lambda_ = lambda_
+
+PDeepPPConfig.register_for_auto_class()

modeling_PDeepPP.py

@@ -5,7 +5,7 @@ from typing import Optional, Tuple, Union
 from transformers.modeling_utils import PreTrainedModel
 from transformers.utils import logging
 
-from configuration_PDeepPP import PDeepPPConfig
+from configuration_pdeeppp import PDeepPPConfig
 
 logger = logging.get_logger(__name__)