get_adapter_state_dict

torchtune.modules.peft.get_adapter_state_dict(state_dict: Dict[str, Any], device: Optional[str] = 'cpu' ) → Dict[str, Any]

返回模型完整状态字典中对应于适配器的子集。假定“lora”和“magnitude”是适配器参数的唯一名称，且状态字典未分片。所有返回的参数都被移至 CPU。

参数:

• state_dict (Dict[str, Any]) – 完整的模型状态字典。

• device (Optional[str]) – 将适配器参数移动到的设备。默认值: ‘cpu’

返回:

模型的状态字典中仅包含适配器参数的子集。

返回类型:

Dict[str, Any]