Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/architkaila/Fine-Tuning-LLMs-for-Medical-Entity-Extraction
/ types & classes
Types & classes
39 in github.com/architkaila/Fine-Tuning-LLMs-for-Medical-Entity-Extraction
⨍
Functions
184
◇
Types & classes
39
↓ 8 callers
Class
AdapterV2Linear
lit_gpt/adapter_v2.py:47
↓ 7 callers
Class
LoRALinear
lit_gpt/lora.py:87
↓ 7 callers
Class
Tokenizer
lit_gpt/tokenizer.py:8
↓ 4 callers
Class
GPT
lit_gpt/lora.py:454
↓ 2 callers
Class
GPT
lit_gpt/adapter_v2.py:62
↓ 2 callers
Class
SavingProxyForStorage
lit_gpt/utils.py:90
↓ 1 callers
Class
Block
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 callers
Class
Block
lit_gpt/lora.py:522
↓ 1 callers
Class
Block
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 callers
Class
Block
lit_gpt/model.py:135
↓ 1 callers
Class
CausalSelfAttention
A modification of `lit_gpt.model.CausalSelfAttention` that adds the attention over the adaption prompt.
lit_gpt/adapter.py:101
↓ 1 callers
Class
CausalSelfAttention
lit_gpt/lora.py:534
↓ 1 callers
Class
CausalSelfAttention
A modification of `lit_gpt.adapter.CausalSelfAttention` that uses the Adapter V2 Linear class
lit_gpt/adapter_v2.py:113
↓ 1 callers
Class
CausalSelfAttention
lit_gpt/model.py:169
↓ 1 callers
Class
CombinedDatasetIterator
lit_gpt/packed_dataset.py:229
↓ 1 callers
Class
IncrementalPyTorchPickler
lit_gpt/utils.py:140
↓ 1 callers
Class
KVCache
lit_gpt/model.py:322
↓ 1 callers
Class
LoRAQKVLinear
lit_gpt/lora.py:153
↓ 1 callers
Class
PackedDatasetIterator
lit_gpt/packed_dataset.py:121
↓ 1 callers
Class
SavingProxyForTensor
lit_gpt/utils.py:118
↓ 1 callers
Class
incremental_save
lit_gpt/utils.py:196
Class
CombinedDataset
lit_gpt/packed_dataset.py:214
Class
Config
lit_gpt/config.py:15
Class
Config
lit_gpt/adapter.py:22
Class
Config
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
Class
Config
lit_gpt/adapter_v2.py:25
Class
GPT
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
Class
GPT
lit_gpt/model.py:16
Class
GptNeoxMLP
lit_gpt/lora.py:579
Class
GptNeoxMLP
lit_gpt/adapter_v2.py:153
Class
GptNeoxMLP
lit_gpt/model.py:264
Class
LLaMAMLP
lit_gpt/lora.py:613
Class
LLaMAMLP
lit_gpt/adapter_v2.py:173
Class
LLaMAMLP
lit_gpt/model.py:278
Class
LoRALayer
lit_gpt/lora.py:62
Class
PackedDataset
lit_gpt/packed_dataset.py:27
Class
PackedDatasetBuilder
lit_gpt/packed_dataset.py:60
Class
QuantizedLinear
lit_gpt/utils.py:49
Class
RMSNorm
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