consistent ignore keys + make private (#8737)
* consistent ignore keys + make private * style * - authorized_missing_keys => _keys_to_ignore_on_load_missing - authorized_unexpected_keys => _keys_to_ignore_on_load_unexpected * move public doc of private attributes to private comment
This commit is contained in:
@@ -216,7 +216,7 @@ class RagPreTrainedModel(PreTrainedModel):
|
||||
"""
|
||||
config_class = RagConfig
|
||||
base_model_prefix = "rag"
|
||||
authorized_missing_keys = [r"position_ids"]
|
||||
_keys_to_ignore_on_load_missing = [r"position_ids"]
|
||||
|
||||
@classmethod
|
||||
def from_pretrained_question_encoder_generator(
|
||||
|
||||
Reference in New Issue
Block a user