Enum TargetModule 

pub enum TargetModule {
    QProj,
    KProj,
    VProj,
    OProj,
    GateProj,
    UpProj,
    DownProj,
}

Target modules for LoRA adaptation.

Specifies which layers in the model should have LoRA applied.

Variants

QProj

Query projection in attention

KProj

Key projection in attention

VProj

Value projection in attention

OProj

Output projection in attention

GateProj

Gate projection in MLP

UpProj

Up projection in MLP

DownProj

Down projection in MLP

Implementations

impl TargetModule

pub const fn name(&self) -> &'static str

Returns the canonical name for this module.

pub fn from_name(name: &str) -> Option<Self>

Parses a module name into a TargetModule.

Trait Implementations

impl Clone for TargetModule

fn clone(&self) -> TargetModule

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for TargetModule

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Hash for TargetModule

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for TargetModule

fn eq(&self, other: &TargetModule) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Eq for TargetModule

impl StructuralPartialEq for TargetModule