Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ZinYY/TreeLoRA
/ types & classes
Types & classes
69 in github.com/ZinYY/TreeLoRA
⨍
Functions
425
◇
Types & classes
69
↓ 8 callers
Class
LoraConfig
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 callers
Class
TIKTOK
utils/model/model_utils.py:22
↓ 6 callers
Class
DataCollator
utils/data/data_collator.py:10
↓ 4 callers
Class
DataCollator
inference/HHH/HHH_data_collator.py:9
↓ 3 callers
Class
KDTreeNode
utils/kd_lora_tree.py:10
↓ 1 callers
Class
AdaptedAttention
This module wraps a LLamaAttention module and injects adaption prompts.
utils/my_peft/tuners/adaption_prompt.py:272
↓ 1 callers
Class
Embedding
utils/my_peft/tuners/lora.py:632
↓ 1 callers
Class
H_Config
inference/HHH/data_process.py:6
↓ 1 callers
Class
KD_LoRA_Tree
utils/kd_lora_tree.py:93
↓ 1 callers
Class
Linear
utils/my_peft/tuners/lora.py:533
↓ 1 callers
Class
Linear8bitLt
utils/my_peft/tuners/lora.py:715
↓ 1 callers
Class
LinearLayer_LoRA
utils/module/lora.py:13
↓ 1 callers
Class
ModulesToSaveWrapper
utils/my_peft/utils/other.py:95
↓ 1 callers
Class
PeftModel
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 callers
Class
PrefixEncoder
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 callers
Class
PromptDataset
utils/data/data_utils.py:223
↓ 1 callers
Class
PromptEmbedding
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 callers
Class
PromptEncoder
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 callers
Class
PromptTuningConfig
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 callers
Class
RankAllocator
The RankAllocator for AdaLoraModel. Paper: https://openreview.net/pdf?id=lq62uWRJjiY Args: config ([`AdaLoraConfig`]): The configura
utils/my_peft/tuners/adalora.py:516
↓ 1 callers
Class
ReplayMemory
Create the empty memory buffer
model/Replay/MbPAplusplus.py:26
↓ 1 callers
Class
SVDLinear
utils/my_peft/tuners/adalora.py:365
↓ 1 callers
Class
SVDLinear8bitLt
utils/my_peft/tuners/adalora.py:452
Class
API
inference/ICL.py:113
Class
AdaLoraConfig
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
Class
AdaLoraLayer
utils/my_peft/tuners/adalora.py:319
Class
AdaLoraModel
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
Class
AdaptionPromptConfig
Stores the configuration of an [`AdaptionPromptModel`].
utils/my_peft/tuners/adaption_prompt.py:105
Class
AdaptionPromptModel
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
Class
AnthropichhrlhfDataset
utils/data/raw_datasets.py:52
Class
CL_Base_Model
model/base_model.py:22
Class
ChatPrediction
utils/data/data_utils.py:83
Class
CompletionPrediction
utils/data/data_utils.py:77
Class
DualPrompt
model/Dynamic_network/DualPrompt.py:58
Class
EWC
model/Regular/EWC.py:12
Class
FlashAttnFunc
utils/flash_attention/triton_flash_att.py:797
Class
FlashAttnKVPackedFunc
utils/flash_attention/triton_flash_att.py:758
Class
FlashAttnQKVPackedFunc
utils/flash_attention/triton_flash_att.py:720
Class
GEM
model/Regular/GEM.py:12
Class
HHH
inference/HHH/data_process.py:14
Class
HideLoRA
model/Regular/HideLoRA.py:13
Class
ImportanceScorer
Importance score calculator for TaSL method. Calculates importance scores for LoRA parameters based on parameter sensitivity. Args:
utils/TaSL/importance_scorer.py:9
Class
LFPT5
model/Replay/LFPT5.py:44
Class
Llama
utils/data/data_utils.py:123
Class
LocalJsonFileDataset
utils/data/raw_datasets.py:80
Class
LoraLayer
utils/my_peft/tuners/lora.py:440
Class
LoraModel
Creates Low Rank Adapter (Lora) model from a pretrained transformers model. Args: model ([`~transformers.PreTrainedModel`]): The mod
utils/my_peft/tuners/lora.py:94
Class
LwF
model/Regular/LwF.py:11
Class
MbPAplusplus
Implements Memory based Parameter Adaptation model
model/Replay/MbPAplusplus.py:109
Class
Message
utils/data/data_utils.py:72
Class
MovingAverage
utils/utils.py:30
Class
OGD
model/Regular/OGD.py:37
Class
O_LoRA
model/Regular/O_LoRA.py:13
Class
PeftConfig
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
Class
PeftConfigMixin
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
Class
PeftModelForCausalLM
Peft model for causal language modeling. Args: model ([`~transformers.PreTrainedModel`]): Base transformer model. peft_confi
utils/my_peft/peft_model.py:624
Class
PeftModelForSeq2SeqLM
Peft model for sequence-to-sequence language modeling. Args: model ([`~transformers.PreTrainedModel`]): Base transformer model.
utils/my_peft/peft_model.py:809
Class
PeftModelForSequenceClassification
Peft model for sequence classification tasks. Args: model ([`~transformers.PreTrainedModel`]): Base transformer model. peft_
utils/my_peft/peft_model.py:436
Class
PeftModelForTokenClassification
Peft model for token classification tasks. Args: model ([`~transformers.PreTrainedModel`]): Base transformer model. peft_con
utils/my_peft/peft_model.py:1008
Class
PeftType
utils/my_peft/utils/config.py:27
Class
PrefixTuningConfig
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
Class
PromptEncoderConfig
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
Class
PromptEncoderReparameterizationType
utils/my_peft/tuners/p_tuning.py:26
Class
PromptLearningConfig
This is the base configuration class to store the configuration of [`PrefixTuning`], [`PromptEncoder`], or [`PromptTuning`]. Args:
utils/my_peft/utils/config.py:152
Class
PromptRawDataset
utils/data/raw_datasets.py:13
Class
PromptTuningInit
utils/my_peft/tuners/prompt_tuning.py:26
Class
TaskType
utils/my_peft/utils/config.py:36
Class
Tree_LoRA
model/Regular/Tree_LoRA.py:16
Class
lora
model/lora.py:9