Update modeling_provence.py

modeling_provence.py

@@ -12,10 +12,14 @@ from torch.utils.data import Dataset
 from torch.nn.utils.rnn import pad_sequence
 from transformers import AutoTokenizer
 from transformers import DebertaV2PreTrainedModel, DebertaV2Model, PretrainedConfig
-from transformers.models.deberta_v2.modeling_deberta_v2 import (
-    StableDropout,
-    ContextPooler,
-)
+try:
+    from transformers.models.deberta_v2.modeling_deberta_v2 import (
+        StableDropout,
+        ContextPooler,
+    )
+except ImportError:
+    from transformers.models.deberta_v2.modeling_deberta_v2 import ContextPooler
+    StableDropout = nn.Dropout
 from transformers.modeling_outputs import ModelOutput