peft

[[来源]]

from gemma import peft

用于 flax.linen 的 PEFT 工具。

所有符号

模块

peft	用于 flax.linen 的 PEFT 工具。

类

peft.IntDense	围绕 nn.Dense 的包装器，添加了量化适配器。
peft.IntEinsum	围绕 nn.Einsum 的包装器，添加了量化适配器。
peft.Interceptor	拦截器的基类。
peft.LoRADense	围绕 nn.Dense 的包装器，添加了 LoRA 适配器。
peft.LoRADenseAdapter	LoRA 模块。
peft.LoRAEinsum	围绕 nn.Einsum 的包装器，添加了 LoRA 适配器。
peft.LoRAEinsumAdapter	LoRA einsum 模块。
peft.ModuleInterceptor	拦截器，捕获所有模块以最终替换它们。
peft.QuantizationMethod	量化方法。
peft.SimulateQuantizedDense	围绕 nn.Dense 的包装器，添加了量化适配器。
peft.SimulateQuantizedEinsum	围绕 nn.Einsum 的包装器，添加了量化适配器。

函数

peft.fuse_params
peft.get_axis_to_reduce_from_einsum_str	返回要缩减的轴。
peft.merge_params	split_params 的逆运算。
peft.quantize	量化给定的参数。
peft.simulate_quantize	量化给定的数组。
peft.split_params	将嵌套树拆分为 2 个树，一个带有 ‘lora’ 分支，另一个没有。
peft.unfuse_params