SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix from_pretrained with default base_model_prefix (#15814)

Browse Source
This commit is contained in:
Sylvain Gugger
2022-02-24 11:43:51 +01:00
committed by GitHub
parent 7f921bcf47
commit d1fcc90abf
3 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
utils/test_module/custom_modeling.py
View File

@@ -7,7 +7,6 @@ from .custom_configuration import CustomConfig, NoSuperInitConfig
class CustomModel(PreTrainedModel):
config_class = CustomConfig
base_model_prefix = "custom"
def __init__(self, config):
super().__init__(config)
@@ -22,7 +21,6 @@ class CustomModel(PreTrainedModel):
class NoSuperInitModel(PreTrainedModel):
config_class = NoSuperInitConfig
base_model_prefix = "custom"
def __init__(self, config):
super().__init__(config)