Update modeling_deberta.py
Browse files- modeling_deberta.py +2 -0
modeling_deberta.py
CHANGED
|
@@ -898,6 +898,8 @@ class DebertaV2PreTrainedModel(PreTrainedModel):
|
|
| 898 |
An abstract class to handle weights initialization and a simple interface for downloading and loading pretrained
|
| 899 |
models.
|
| 900 |
"""
|
|
|
|
|
|
|
| 901 |
supports_gradient_checkpointing = True
|
| 902 |
|
| 903 |
def _init_weights(self, module):
|
|
|
|
| 898 |
An abstract class to handle weights initialization and a simple interface for downloading and loading pretrained
|
| 899 |
models.
|
| 900 |
"""
|
| 901 |
+
config_class = DebertaV2Config
|
| 902 |
+
base_model_prefix = "deberta"
|
| 903 |
supports_gradient_checkpointing = True
|
| 904 |
|
| 905 |
def _init_weights(self, module):
|