Update modeling_perceive_sampler.py
Browse files
modeling_perceive_sampler.py
CHANGED
|
@@ -938,6 +938,8 @@ class BertLMHeadModel(BertPreTrainedModel):
|
|
| 938 |
r"position_ids", r"predictions.decoder.bias"
|
| 939 |
]
|
| 940 |
|
|
|
|
|
|
|
| 941 |
def __init__(self, config):
|
| 942 |
super().__init__(config)
|
| 943 |
|
|
|
|
| 938 |
r"position_ids", r"predictions.decoder.bias"
|
| 939 |
]
|
| 940 |
|
| 941 |
+
_no_split_modules = ["InternLMDecoderLayer"]
|
| 942 |
+
|
| 943 |
def __init__(self, config):
|
| 944 |
super().__init__(config)
|
| 945 |
|