Function save_configuration

modelscope/utils/checkpoint.py:574–581

(target_folder, config: Dict)

def save_configuration(target_folder, config: Dict):
    if isinstance(config, Config):
        config = config.to_dict()
    if ConfigFields.pipeline not in config:
        config[ConfigFields.pipeline] = {'type': config[ConfigFields.task]}
    cfg_str = json.dumps(config, indent=4, cls=JSONIteratorEncoder)
    config_file = os.path.join(target_folder, ModelFile.CONFIGURATION)
    storage.write(cfg_str.encode(), config_file)


def save_pretrained(model,