mindformers.AutoConfig
- class mindformers.AutoConfig[源代码]
这是一个通用的配置类,当使用 from_pretrained() 类方法时,它会自动实例化模型Config类,并返回。 这个类不能直接使用 __init__() 实例化(会抛出异常)。
样例:
>>> from mindformers import AutoConfig >>> # 1) instantiates a config by yaml model name >>> config_a = AutoConfig.from_pretrained('configs/glm4/predict_glm4_9b_chat.yaml') >>> # 2) instantiates a config by json file >>> config_b = AutoConfig.from_pretrained('./config.json') >>> # 3) instantiates a config by directory containing a configuration json file >>> config_c = AutoConfig.from_pretrained('./dir/') >>> # 4) instantiates a config by model_id from modelers.cn >>> config_d = AutoConfig.from_pretrained('MindSpore-Lab/glm2_6b')
- classmethod from_pretrained(yaml_name_or_path, **kwargs)[源代码]
从yaml文件、json文件、文件夹或魔乐社区读取配置信息,实例化为模型Config类,并返回。
警告
这个API正处于实验阶段,在下一个版本中可能会有一些突破性的变化。
- 参数:
yaml_name_or_path (str) - yaml文件路径、json文件路径、包含config.json文件的文件夹路径、或魔乐社区上的model_id。后三者为实验特性。
kwargs (Dict[str, Any], 可选) - 传入的配置信息将会覆盖从yaml_name_or_path读取到的配置信息。
- 返回:
一个继承自PretrainedConfig类的模型Config实例。