Update configuration_protst.py
Browse files- configuration_protst.py +1 -1
configuration_protst.py
CHANGED
|
@@ -20,7 +20,7 @@ class ProtSTConfig(PretrainedConfig):
|
|
| 20 |
Dictionary of configuration options used to initialize [`BertForPubMed`].
|
| 21 |
```"""
|
| 22 |
|
| 23 |
-
|
| 24 |
|
| 25 |
def __init__(
|
| 26 |
self,
|
|
|
|
| 20 |
Dictionary of configuration options used to initialize [`BertForPubMed`].
|
| 21 |
```"""
|
| 22 |
|
| 23 |
+
model_type = "protst"
|
| 24 |
|
| 25 |
def __init__(
|
| 26 |
self,
|