转换

Medusa 转换/恢复工具。

函数

convert_to_medusa_model

根据config将模型转换为medusa模型。

restore_medusa_model

用于将先前转换的模型恢复为medusa模型的函数。
convert_to_medusa_model(model, config)

根据config将模型转换为medusa模型。

Parameters:
Return type:

元组[模块, 字典[字符串, 任意类型]]

restore_medusa_model(model, config, metadata)

用于将先前转换的模型恢复为medusa模型的函数。

Parameters:

  • 模型 (模块) –

  • config (MedusaConfig) –

  • metadata (Dict[str, Any]) –

Return type:

模块