MCPcopy Create free account

hub / github.com/ZinYY/TreeLoRA / types & classes

Types & classes69 in github.com/ZinYY/TreeLoRA

↓ 8 callersClassLoraConfig
This is the configuration class to store the configuration of a [`LoraModel`]. Args: r (`int`): Lora attention dimension. ta
utils/my_peft/tuners/lora.py:44
↓ 7 callersClassTIKTOK
utils/model/model_utils.py:22
↓ 6 callersClassDataCollator
utils/data/data_collator.py:10
↓ 4 callersClassDataCollator
inference/HHH/HHH_data_collator.py:9
↓ 3 callersClassKDTreeNode
utils/kd_lora_tree.py:10
↓ 1 callersClassAdaptedAttention
This module wraps a LLamaAttention module and injects adaption prompts.
utils/my_peft/tuners/adaption_prompt.py:272
↓ 1 callersClassEmbedding
utils/my_peft/tuners/lora.py:632
↓ 1 callersClassH_Config
inference/HHH/data_process.py:6
↓ 1 callersClassKD_LoRA_Tree
utils/kd_lora_tree.py:93
↓ 1 callersClassLinear
utils/my_peft/tuners/lora.py:533
↓ 1 callersClassLinear8bitLt
utils/my_peft/tuners/lora.py:715
↓ 1 callersClassLinearLayer_LoRA
utils/module/lora.py:13
↓ 1 callersClassModulesToSaveWrapper
utils/my_peft/utils/other.py:95
↓ 1 callersClassPeftModel
Base model encompassing various Peft methods. Args: model ([`~transformers.PreTrainedModel`]): The base transformer model used for P
utils/my_peft/peft_model.py:64
↓ 1 callersClassPrefixEncoder
r""" The `torch.nn` model to encode the prefix. Args: config ([`PrefixTuningConfig`]): The configuration of the prefix encoder.
utils/my_peft/tuners/prefix_tuning.py:49
↓ 1 callersClassPromptDataset
utils/data/data_utils.py:223
↓ 1 callersClassPromptEmbedding
The model to encode virtual tokens into prompt embeddings. Args: config ([`PromptTuningConfig`]): The configuration of the prompt em
utils/my_peft/tuners/prompt_tuning.py:65
↓ 1 callersClassPromptEncoder
The prompt encoder network that is used to generate the virtual token embeddings for p-tuning. Args: config ([`PromptEncoderConfig`]
utils/my_peft/tuners/p_tuning.py:67
↓ 1 callersClassPromptTuningConfig
This is the configuration class to store the configuration of a [`PromptEmbedding`]. Args: prompt_tuning_init (Union[[`PromptTuningI
utils/my_peft/tuners/prompt_tuning.py:32
↓ 1 callersClassRankAllocator
The RankAllocator for AdaLoraModel. Paper: https://openreview.net/pdf?id=lq62uWRJjiY Args: config ([`AdaLoraConfig`]): The configura
utils/my_peft/tuners/adalora.py:516
↓ 1 callersClassReplayMemory
Create the empty memory buffer
model/Replay/MbPAplusplus.py:26
↓ 1 callersClassSVDLinear
utils/my_peft/tuners/adalora.py:365
↓ 1 callersClassSVDLinear8bitLt
utils/my_peft/tuners/adalora.py:452
ClassAPI
inference/ICL.py:113
ClassAdaLoraConfig
This is the configuration class to store the configuration of a [`~peft.AdaLora`]. Args: target_r (`int`): The target average rank o
utils/my_peft/tuners/adalora.py:36
ClassAdaLoraLayer
utils/my_peft/tuners/adalora.py:319
ClassAdaLoraModel
Creates AdaLoRA (Adaptive LoRA) model from a pretrained transformers model. Paper: https://openreview.net/pdf?id=lq62uWRJjiY Args:
utils/my_peft/tuners/adalora.py:68
ClassAdaptionPromptConfig
Stores the configuration of an [`AdaptionPromptModel`].
utils/my_peft/tuners/adaption_prompt.py:105
ClassAdaptionPromptModel
Implements adaption prompts as described in https://arxiv.org/pdf/2303.16199.pdf. The top L attention modules are replaced with AdaptedAtten
utils/my_peft/tuners/adaption_prompt.py:134
ClassAnthropichhrlhfDataset
utils/data/raw_datasets.py:52
ClassCL_Base_Model
model/base_model.py:22
ClassChatPrediction
utils/data/data_utils.py:83
ClassCompletionPrediction
utils/data/data_utils.py:77
ClassDualPrompt
model/Dynamic_network/DualPrompt.py:58
ClassEWC
model/Regular/EWC.py:12
ClassFlashAttnFunc
utils/flash_attention/triton_flash_att.py:797
ClassFlashAttnKVPackedFunc
utils/flash_attention/triton_flash_att.py:758
ClassFlashAttnQKVPackedFunc
utils/flash_attention/triton_flash_att.py:720
ClassGEM
model/Regular/GEM.py:12
ClassHHH
inference/HHH/data_process.py:14
ClassHideLoRA
model/Regular/HideLoRA.py:13
ClassImportanceScorer
Importance score calculator for TaSL method. Calculates importance scores for LoRA parameters based on parameter sensitivity. Args:
utils/TaSL/importance_scorer.py:9
ClassLFPT5
model/Replay/LFPT5.py:44
ClassLlama
utils/data/data_utils.py:123
ClassLocalJsonFileDataset
utils/data/raw_datasets.py:80
ClassLoraLayer
utils/my_peft/tuners/lora.py:440
ClassLoraModel
Creates Low Rank Adapter (Lora) model from a pretrained transformers model. Args: model ([`~transformers.PreTrainedModel`]): The mod
utils/my_peft/tuners/lora.py:94
ClassLwF
model/Regular/LwF.py:11
ClassMbPAplusplus
Implements Memory based Parameter Adaptation model
model/Replay/MbPAplusplus.py:109
ClassMessage
utils/data/data_utils.py:72
ClassMovingAverage
utils/utils.py:30
ClassOGD
model/Regular/OGD.py:37
ClassO_LoRA
model/Regular/O_LoRA.py:13
ClassPeftConfig
This is the base configuration class to store the configuration of a [`PeftModel`]. Args: peft_type (Union[[`~peft.utils.config.Peft
utils/my_peft/utils/config.py:136
ClassPeftConfigMixin
r""" This is the base configuration class for PEFT adapter models. It contains all the methods that are common to all PEFT adapter models. Thi
utils/my_peft/utils/config.py:44
ClassPeftModelForCausalLM
Peft model for causal language modeling. Args: model ([`~transformers.PreTrainedModel`]): Base transformer model. peft_confi
utils/my_peft/peft_model.py:624
ClassPeftModelForSeq2SeqLM
Peft model for sequence-to-sequence language modeling. Args: model ([`~transformers.PreTrainedModel`]): Base transformer model.
utils/my_peft/peft_model.py:809
ClassPeftModelForSequenceClassification
Peft model for sequence classification tasks. Args: model ([`~transformers.PreTrainedModel`]): Base transformer model. peft_
utils/my_peft/peft_model.py:436
ClassPeftModelForTokenClassification
Peft model for token classification tasks. Args: model ([`~transformers.PreTrainedModel`]): Base transformer model. peft_con
utils/my_peft/peft_model.py:1008
ClassPeftType
utils/my_peft/utils/config.py:27
ClassPrefixTuningConfig
This is the configuration class to store the configuration of a [`PrefixEncoder`]. Args: encoder_hidden_size (`int`): The hidden siz
utils/my_peft/tuners/prefix_tuning.py:25
ClassPromptEncoderConfig
This is the configuration class to store the configuration of a [`PromptEncoder`]. Args: encoder_reparameterization_type (Union[[`Pr
utils/my_peft/tuners/p_tuning.py:32
ClassPromptEncoderReparameterizationType
utils/my_peft/tuners/p_tuning.py:26
ClassPromptLearningConfig
This is the base configuration class to store the configuration of [`PrefixTuning`], [`PromptEncoder`], or [`PromptTuning`]. Args:
utils/my_peft/utils/config.py:152
ClassPromptRawDataset
utils/data/raw_datasets.py:13
ClassPromptTuningInit
utils/my_peft/tuners/prompt_tuning.py:26
ClassTaskType
utils/my_peft/utils/config.py:36
ClassTree_LoRA
model/Regular/Tree_LoRA.py:16
Classlora
model/lora.py:9