Skip to content

GitLab

Unverified Commit f1bb6f08 authored by Allan Lin's avatar Allan Lin Committed by GitHub
Browse files

Fix load tf alias in Albert. (#13159)

parent 0b54046f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/models/albert/modeling_albert.py
View file @ f1bb6f08
...@@ -494,6 +494,7 @@ class AlbertPreTrainedModel(PreTrainedModel): ...@@ -494,6 +494,7 @@ class AlbertPreTrainedModel(PreTrainedModel):
""" """
config_class = AlbertConfig config_class = AlbertConfig
load_tf_weights = load_tf_weights_in_albert
base_model_prefix = "albert" base_model_prefix = "albert"
_keys_to_ignore_on_load_missing = [r"position_ids"] _keys_to_ignore_on_load_missing = [r"position_ids"]
...@@ -623,7 +624,6 @@ ALBERT_INPUTS_DOCSTRING = r""" ...@@ -623,7 +624,6 @@ ALBERT_INPUTS_DOCSTRING = r"""
class AlbertModel(AlbertPreTrainedModel): class AlbertModel(AlbertPreTrainedModel):
config_class = AlbertConfig config_class = AlbertConfig
load_tf_weights = load_tf_weights_in_albert
base_model_prefix = "albert" base_model_prefix = "albert"
def __init__(self, config, add_pooling_layer=True): def __init__(self, config, add_pooling_layer=True):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment