Update configuration_mixin.py
Browse files- configuration_mixin.py +2 -2
configuration_mixin.py
CHANGED
@@ -31,8 +31,8 @@ class MixinConfig(PretrainedConfig):
|
|
31 |
def from_pretrained(cls, pretrained_model_name_or_path: Union[str, os.PathLike], **kwargs) -> 'PretrainedConfig':
|
32 |
config_dict, kwargs = cls.get_config_dict(pretrained_model_name_or_path, **kwargs)
|
33 |
|
34 |
-
if '
|
35 |
-
config_dict = config_dict['
|
36 |
|
37 |
if 'model_type' in config_dict and hasattr(cls, 'model_type') and config_dict['model_type'] != cls.model_type:
|
38 |
logger.warning(
|
|
|
31 |
def from_pretrained(cls, pretrained_model_name_or_path: Union[str, os.PathLike], **kwargs) -> 'PretrainedConfig':
|
32 |
config_dict, kwargs = cls.get_config_dict(pretrained_model_name_or_path, **kwargs)
|
33 |
|
34 |
+
if 'mixin_config' in config_dict:
|
35 |
+
config_dict = config_dict['mixin_config']
|
36 |
|
37 |
if 'model_type' in config_dict and hasattr(cls, 'model_type') and config_dict['model_type'] != cls.model_type:
|
38 |
logger.warning(
|