peft.IntEinsum

class gemma.peft.IntEinsum(

*,

wrapped: flax.linen.linear.Einsum,

dtype: numpy.dtype = <class 'jax.numpy.int4'>,

parent: flax.linen.module.Module | flax.core.scope.Scope | flax.linen.module._Sentinel | None = <flax.linen.module._Sentinel object>,

name: str | None = None,

)

Bases: flax.linen.module.Module

围绕 nn.Einsum 的包装器，添加了一个量化适配器。

wrapped: flax.linen.linear.Einsum

dtype

别名 jax.numpy.int4

process_einsum_str(einsum_str: str) → str

处理 einsum 字符串。

name: str | None = None

parent: flax.linen.module.Module | flax.core.scope.Scope | flax.linen.module._Sentinel | None = None

scope: Scope | None = None