Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Pints-AI/1.5-Pints
/ types & classes
Types & classes
173 in github.com/Pints-AI/1.5-Pints
⨍
Functions
589
◇
Types & classes
173
↓ 11 callers
Class
Tokenizer
lit_gpt/tokenizer.py:12
↓ 8 callers
Class
AdapterV2Linear
lit_gpt/adapter_v2.py:49
↓ 8 callers
Class
SFTMultiTurnDataset
A multiturn version of `SFTDataset`, which is an in-memory dataset for supervised finetuning with `input_ids` and `labels`. Args: data: A
lit_gpt/datamodules/sft_multiturn_dataset_base.py:20
↓ 7 callers
Class
LazyTensor
tokenizer/convert/convert.py:645
↓ 7 callers
Class
LoRALinear
lit_gpt/lora.py:87
↓ 7 callers
Class
UnquantizedTensor
tokenizer/convert/convert.py:598
↓ 6 callers
Class
SFTDataset
An in-memory dataset for supervised finetuning with `input_ids` and `labels`. Args: data: A list of samples (dicts). The target/label mus
lit_gpt/datamodules/sft_dataset_base.py:8
↓ 5 callers
Class
CycleIterator
An iterator that cycles through an iterable indefinitely. Example: >>> iterator = CycleIterator([1, 2, 3]) >>> [next(iterator) fo
lit_gpt/utils.py:413
↓ 5 callers
Class
GPT
lit_gpt/model.py:43
↓ 4 callers
Class
LazyStorageKind
tokenizer/convert/convert.py:769
↓ 4 callers
Class
ModelPlus
tokenizer/convert/convert.py:674
↓ 4 callers
Class
UnquantizedDataType
tokenizer/convert/convert.py:65
↓ 3 callers
Class
ChatML
lit_gpt/prompts.py:44
↓ 3 callers
Class
PackedDatasetBuilder
A class designed to manage the packing and storage of large arrays into binary files with a specified chunk size. This class handles the div
lit_gpt/packed_dataset.py:81
↓ 3 callers
Class
Params
tokenizer/convert/convert.py:157
↓ 3 callers
Class
lazy_load
lit_gpt/utils_old.py:218
↓ 2 callers
Class
Deita
Deita data module for supervised finetuning.
lit_gpt/datamodules/deita.py:39
↓ 2 callers
Class
MetaMathQA
MetaMathQA data module for supervised finetuning.
lit_gpt/datamodules/meta_math_qa.py:34
↓ 2 callers
Class
OutputFile
tokenizer/convert/convert.py:969
↓ 2 callers
Class
SavingProxyForStorage
lit_gpt/utils.py:106
↓ 2 callers
Class
SlimOrcaIDK
SlimOrcaIDK data module for supervised finetuning.
lit_gpt/datamodules/slim_orca_idontknow.py:36
↓ 2 callers
Class
Style
lit_gpt/datamodules/base_test.py:15
↓ 2 callers
Class
incremental_save
lit_gpt/utils_old.py:365
↓ 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:125
↓ 1 callers
Class
Block
lit_gpt/lora.py:600
↓ 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:113
↓ 1 callers
Class
Block
lit_gpt/model.py:245
↓ 1 callers
Class
BpeVocab
tokenizer/convert/convert.py:337
↓ 1 callers
Class
Capybara
Capybara data module for supervised finetuning.
lit_gpt/datamodules/capybara.py:38
↓ 1 callers
Class
CausalSelfAttention
A modification of `lit_gpt.model.CausalSelfAttention` that adds the attention over the adaption prompt.
lit_gpt/adapter.py:167
↓ 1 callers
Class
CausalSelfAttention
lit_gpt/lora.py:612
↓ 1 callers
Class
CausalSelfAttention
lit_gpt/adapter_v2.py:129
↓ 1 callers
Class
CausalSelfAttention
lit_gpt/model.py:283
↓ 1 callers
Class
CodeLlama
lit_gpt/prompts.py:260
↓ 1 callers
Class
CombinedDatasetIterator
lit_gpt/packed_dataset.py:296
↓ 1 callers
Class
Default
lit_gpt/prompts.py:36
↓ 1 callers
Class
ErrorCountHandler
This is to be attached to our logging interface. Useful at the end of dataset preparation, to tell user how many errors are there.
prepare_dataset/preparer.py:39
↓ 1 callers
Class
EvalArgs
Evaluation-related arguments
lit_gpt/args.py:59
↓ 1 callers
Class
Falcon
lit_gpt/prompts.py:160
↓ 1 callers
Class
FreeWilly2
lit_gpt/prompts.py:235
↓ 1 callers
Class
FusedCrossEntropyLoss
lit_gpt/fused_cross_entropy.py:140
↓ 1 callers
Class
Gemma
lit_gpt/prompts.py:300
↓ 1 callers
Class
HfVocab
tokenizer/convert/convert.py:458
↓ 1 callers
Class
IncrementalPyTorchPickler
lit_gpt/utils_old.py:309
↓ 1 callers
Class
IncrementalPyTorchPickler
lit_gpt/utils.py:172
↓ 1 callers
Class
LazyLoadingUnpickler
lit_gpt/utils_old.py:194
↓ 1 callers
Class
LazyStorage
tokenizer/convert/convert.py:774
↓ 1 callers
Class
LazyUnpickler
tokenizer/convert/convert.py:780
↓ 1 callers
Class
Llama2
lit_gpt/prompts.py:221
↓ 1 callers
Class
Llama2FunctionCalling
lit_gpt/prompts.py:186
↓ 1 callers
Class
LlamaInstruct
LlamaInstruct data module for supervised finetuning.
lit_gpt/datamodules/llama_instruct.py:31
↓ 1 callers
Class
LoRAQKVLinear
lit_gpt/lora.py:157
↓ 1 callers
Class
MockTokenizer
A dummy tokenizer that encodes each character as its ASCII code.
conftest.py:24
↓ 1 callers
Class
MovingAverage
finetune/full.py:57
↓ 1 callers
Class
Nectar
dpo/adapters/nectar.py:24
↓ 1 callers
Class
NotYetLoadedTensor
lit_gpt/utils_old.py:94
↓ 1 callers
Class
NousResearch
lit_gpt/prompts.py:250
↓ 1 callers
Class
PackedDataset
lit_gpt/packed_dataset.py:40
↓ 1 callers
Class
PackedDatasetIterator
lit_gpt/packed_dataset.py:179
↓ 1 callers
Class
Phi1
lit_gpt/prompts.py:269
↓ 1 callers
Class
Phi2
lit_gpt/prompts.py:284
↓ 1 callers
Class
Platypus
lit_gpt/prompts.py:245
↓ 1 callers
Class
Q8_0QuantizedDataType
tokenizer/convert/convert.py:90
↓ 1 callers
Class
SavingProxyForStorage
lit_gpt/utils_old.py:267
↓ 1 callers
Class
SavingProxyForTensor
lit_gpt/utils_old.py:295
↓ 1 callers
Class
SavingProxyForTensor
lit_gpt/utils.py:140
↓ 1 callers
Class
SentencePieceVocab
tokenizer/convert/convert.py:395
↓ 1 callers
Class
SlimOrcaDedup
SlimOrcaDedup data module for supervised finetuning.
lit_gpt/datamodules/slim_orca_dedup.py:38
↓ 1 callers
Class
SpeedMonitorBase
Logs the training throughput and utilization. +-------------------------------------+-----------------------------------------------------------+
lit_gpt/speed_monitor.py:153
↓ 1 callers
Class
Splitter
A class to manage splitting based on a specified ratio, indicating the frequency of False outcomes. A ratio of 0.9 will return 1 True for eve
util/splitter.py:1
↓ 1 callers
Class
StableCode
lit_gpt/prompts.py:255
↓ 1 callers
Class
StableLMAlpha
lit_gpt/prompts.py:103
↓ 1 callers
Class
StableLMZephyr
lit_gpt/prompts.py:122
↓ 1 callers
Class
StandardParquetPreparer
prepare_dataset/standard_parquet.py:17
↓ 1 callers
Class
TinyLlama
lit_gpt/prompts.py:289
↓ 1 callers
Class
TogetherComputerChat
lit_gpt/prompts.py:127
↓ 1 callers
Class
TogetherComputerInstruct
lit_gpt/prompts.py:141
↓ 1 callers
Class
TrainArgs
Training-related arguments
lit_gpt/args.py:8
↓ 1 callers
Class
UltraChat
UltraChat data module for supervised finetuning.
lit_gpt/datamodules/ultrachat_200k.py:38
↓ 1 callers
Class
Vicuna
lit_gpt/prompts.py:177
↓ 1 callers
Class
VocabFactory
tokenizer/convert/convert.py:1287
↓ 1 callers
Class
WizardLMEvolInstructV2
WizardLMEvolInstructV2 data module for supervised finetuning.
lit_gpt/datamodules/wizardlm_evol_instruct_v2.py:40
↓ 1 callers
Class
incremental_save
lit_gpt/utils.py:228
Class
Alpaca
lit_gpt/prompts.py:70
Class
ApplyRotaryEmb
lit_gpt/fused_rotary_embedding.py:10
Class
CapybaraMessage
lit_gpt/datamodules/capybara.py:27
Class
CapybaraRow
lit_gpt/datamodules/capybara.py:32
Class
CombinedDataset
lit_gpt/packed_dataset.py:283
Class
Config
Config subclasses ConfigBase to add on things that is not required outside of training, so that data processing can be done. Independent
lit_gpt/config.py:7
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:458
Class
Config
lit_gpt/adapter_v2.py:27
Class
ConfigBase
lit_gpt/config_base.py:12
Class
DataArguments
finetune/custom_dataclasses.py:16
Class
DataCollatorForCausalLM
finetune/datacollators.py:246
Class
DataCollatorForCausalLMDecoderOnlyChatML
This DataCollator is purpose-made for ChatML templates. Referenced: https://medium.com/@xuebinbin12/fine-tuning-chat-based-llm-with-multi-tur
finetune/datacollators.py:14
Class
DataModule
Base class for all data modules in LitGPT.
lit_gpt/datamodules/base.py:16
Class
DataType
tokenizer/convert/convert.py:55
Class
DatasetAdapter
dpo/adapters/base.py:18
Class
DatasetPreparer
prepare_dataset/preparer.py:54
next →
1–100 of 173, ranked by callers