MCPcopy Create free account

hub / github.com/architkaila/Fine-Tuning-LLMs-for-Medical-Entity-Extraction / types & classes

Types & classes39 in github.com/architkaila/Fine-Tuning-LLMs-for-Medical-Entity-Extraction

↓ 8 callersClassAdapterV2Linear
lit_gpt/adapter_v2.py:47
↓ 7 callersClassLoRALinear
lit_gpt/lora.py:87
↓ 7 callersClassTokenizer
lit_gpt/tokenizer.py:8
↓ 4 callersClassGPT
lit_gpt/lora.py:454
↓ 2 callersClassGPT
lit_gpt/adapter_v2.py:62
↓ 2 callersClassSavingProxyForStorage
lit_gpt/utils.py:90
↓ 1 callersClassBlock
The implementation is identical to `lit_gpt.model.Block` with the exception that we replace the attention layer where adaption is implemented.
lit_gpt/adapter.py:85
↓ 1 callersClassBlock
lit_gpt/lora.py:522
↓ 1 callersClassBlock
The implementation is identical to `lit_gpt.model.Block` with the exception that we replace the attention layer where adaption is implemented.
lit_gpt/adapter_v2.py:97
↓ 1 callersClassBlock
lit_gpt/model.py:135
↓ 1 callersClassCausalSelfAttention
A modification of `lit_gpt.model.CausalSelfAttention` that adds the attention over the adaption prompt.
lit_gpt/adapter.py:101
↓ 1 callersClassCausalSelfAttention
lit_gpt/lora.py:534
↓ 1 callersClassCausalSelfAttention
A modification of `lit_gpt.adapter.CausalSelfAttention` that uses the Adapter V2 Linear class
lit_gpt/adapter_v2.py:113
↓ 1 callersClassCausalSelfAttention
lit_gpt/model.py:169
↓ 1 callersClassCombinedDatasetIterator
lit_gpt/packed_dataset.py:229
↓ 1 callersClassIncrementalPyTorchPickler
lit_gpt/utils.py:140
↓ 1 callersClassKVCache
lit_gpt/model.py:322
↓ 1 callersClassLoRAQKVLinear
lit_gpt/lora.py:153
↓ 1 callersClassPackedDatasetIterator
lit_gpt/packed_dataset.py:121
↓ 1 callersClassSavingProxyForTensor
lit_gpt/utils.py:118
↓ 1 callersClassincremental_save
lit_gpt/utils.py:196
ClassCombinedDataset
lit_gpt/packed_dataset.py:214
ClassConfig
lit_gpt/config.py:15
ClassConfig
lit_gpt/adapter.py:22
ClassConfig
Args: r: rank of the weight update matrices. To make sense of using LoRA the rank should be smaller than the rank of the weig
lit_gpt/lora.py:427
ClassConfig
lit_gpt/adapter_v2.py:25
ClassGPT
The implementation is identical to `lit_gpt.model.GPT` with the exception that the `Block` saves the layer index and passes it down to the attenti
lit_gpt/adapter.py:27
ClassGPT
lit_gpt/model.py:16
ClassGptNeoxMLP
lit_gpt/lora.py:579
ClassGptNeoxMLP
lit_gpt/adapter_v2.py:153
ClassGptNeoxMLP
lit_gpt/model.py:264
ClassLLaMAMLP
lit_gpt/lora.py:613
ClassLLaMAMLP
lit_gpt/adapter_v2.py:173
ClassLLaMAMLP
lit_gpt/model.py:278
ClassLoRALayer
lit_gpt/lora.py:62
ClassPackedDataset
lit_gpt/packed_dataset.py:27
ClassPackedDatasetBuilder
lit_gpt/packed_dataset.py:60
ClassQuantizedLinear
lit_gpt/utils.py:49
ClassRMSNorm
Root Mean Square Layer Normalization. Derived from https://github.com/bzhangGo/rmsnorm/blob/master/rmsnorm_torch.py. BSD 3-Clause License: ht
lit_gpt/rmsnorm.py:4