disable_adapter

torchtune.modules.peft.disable_adapter(model: Module) → Generator[None, None, None]

临时禁用模型中的适配器。例如，这可以用于 DPO 中，将 LoRA 适配器视为策略模型，并禁用它以将基础模型视为参考模型。

此上下文管理器遍历所提供的神经网络模型中的所有模块，如果模块具有可调用的 adapter_params 属性和 disabled 属性，则将其 disabled 设置为 True。然后，控制权交还给调用者。当退出上下文管理器时，它会将所有临时禁用的模块的 disabled 重新设置为 False。

参数:

model (nn.Module) – 要临时禁用其适配器的模型。

产生:

None – 此函数将控制权交还给调用者，同时禁用适配器。

示例

>>> with disable_adapter(model):
...     # Perform operations with adapters disabled
...     pass