CLLBJ16 commited on
Commit
c1ea139
·
verified ·
1 Parent(s): c1b40bf

Update configuration_mixin.py

Browse files
Files changed (1) hide show
  1. 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 'mixin_layers_config' in config_dict:
35
- config_dict = config_dict['mixin_layers_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(
 
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(