模型合并
PEFT 提供了一些用于 合并 LoRA 适配器 的内部实用工具,使用 TIES 和 DARE 方法。
peft.utils.merge_utils.prune
< 源代码 >( tensor: 张量 density: 浮点数 method: 字面量 rescale: 布尔值 = False ) → torch.Tensor
根据 method
修剪任务张量的值。
peft.utils.merge_utils.calculate_majority_sign_mask
< 源代码 >( tensor: 张量 method: 字面量 = 'total' ) → torch.Tensor
获取跨任务张量的多数符号掩码。任务张量在维度 0 上堆叠。
peft.utils.merge_utils.disjoint_merge
< 源代码 >( task_tensors: 张量 majority_sign_mask: 张量 ) → torch.Tensor
合并后的张量。
使用不相交合并的方式合并任务张量。
peft.utils.merge_utils.task_arithmetic
< source >( task_tensors: List weights: Tensor ) → torch.Tensor
使用 task arithmetic
合并任务张量。
peft.utils.merge_utils.ties
< source >( task_tensors: List weights: Tensor density: float majority_sign_method: Literal = 'total' ) → torch.Tensor
使用 ties
合并任务张量。
peft.utils.merge_utils.dare_linear
< source >( task_tensors: List weights: Tensor density: float ) → torch.Tensor
使用 dare linear
合并任务张量。