PEFT documentation
LyCORIS
LyCORIS
LyCORIS (Lora beYond Conventional methods, Other Rank adaptation Implementations for Stable diffusion) sont des adaptateurs de décomposition de matrices semblables à LoRA qui modifient la couche d’attention croisée du UNet. Les méthodes LoHa et LoKr héritent des classes Lycoris présentées ici.
LycorisConfig
classe peft.tuners.lycoris_utils.LycorisConfig
< source >( task_type: Optional[Union[str, TaskType]] = Nonepeft_type: Optional[Union[str, PeftType]] = Noneauto_mapping: Optional[dict] = Nonepeft_version: Optional[str] = Nonebase_model_name_or_path: Optional[str] = Nonerevision: Optional[str] = Noneinference_mode: bool = Falserank_pattern: Optional[dict] = <factory>alpha_pattern: Optional[dict] = <factory> )
Configuration de base pour les adaptateurs de type LyCORIS
LycorisLayer
Couche de base pour les adaptateurs de type LyCORIS
merge
< source >( safe_merge: bool = Falseadapter_names: Optional[list[str]] = None )
Paramètres
- safe_merge (
bool, facultatif) — SiTrue, l’opération de fusion est effectuée sur une copie des poids d’origine et vérifie la présence de NaN avant de fusionner les poids. Cela est utile si vous souhaitez vérifier si l’opération de fusion produira des NaN. Valeur par défaut :False. - adapter_names (
List[str], facultatif) — La liste des noms d’adaptateurs à fusionner. SiNone, tous les adaptateurs actifs seront fusionnés. Valeur par défaut :None.
Fusionner les poids de l'adaptateur actif dans les poids de base
Cette méthode désintègre toutes les couches d'adaptateur fusionnées à partir des poids de base.
LycorisTuner
classe peft.tuners.lycoris_utils.LycorisTuner
< source >( modelpeft_config: Union[PeftConfig, dict[str, PeftConfig]]adapter_name: strlow_cpu_mem_usage: bool = Falsestate_dict: Optional[dict[str, torch.Tensor]] = None )
Paramètres
- model (
torch.nn.Module) — Le modèle à adapter. - config (LoraConfig) — La configuration du modèle Lora.
- adapter_name (
str) — Le nom de l'adaptateur, par défaut"default". - low_cpu_mem_usage (
bool,optional, par défautFalse) — Crée des poids d'adaptateurs vides sur l'appareil méta. Utile pour accélérer le processus de chargement.
Un tuner de base pour les adaptateurs de type LyCORIS