MCPcopy Create free account

hub / github.com/Pints-AI/1.5-Pints / types & classes

Types & classes173 in github.com/Pints-AI/1.5-Pints

↓ 11 callersClassTokenizer
lit_gpt/tokenizer.py:12
↓ 8 callersClassAdapterV2Linear
lit_gpt/adapter_v2.py:49
↓ 8 callersClassSFTMultiTurnDataset
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 callersClassLazyTensor
tokenizer/convert/convert.py:645
↓ 7 callersClassLoRALinear
lit_gpt/lora.py:87
↓ 7 callersClassUnquantizedTensor
tokenizer/convert/convert.py:598
↓ 6 callersClassSFTDataset
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 callersClassCycleIterator
An iterator that cycles through an iterable indefinitely. Example: >>> iterator = CycleIterator([1, 2, 3]) >>> [next(iterator) fo
lit_gpt/utils.py:413
↓ 5 callersClassGPT
lit_gpt/model.py:43
↓ 4 callersClassLazyStorageKind
tokenizer/convert/convert.py:769
↓ 4 callersClassModelPlus
tokenizer/convert/convert.py:674
↓ 4 callersClassUnquantizedDataType
tokenizer/convert/convert.py:65
↓ 3 callersClassChatML
lit_gpt/prompts.py:44
↓ 3 callersClassPackedDatasetBuilder
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 callersClassParams
tokenizer/convert/convert.py:157
↓ 3 callersClasslazy_load
lit_gpt/utils_old.py:218
↓ 2 callersClassDeita
Deita data module for supervised finetuning.
lit_gpt/datamodules/deita.py:39
↓ 2 callersClassMetaMathQA
MetaMathQA data module for supervised finetuning.
lit_gpt/datamodules/meta_math_qa.py:34
↓ 2 callersClassOutputFile
tokenizer/convert/convert.py:969
↓ 2 callersClassSavingProxyForStorage
lit_gpt/utils.py:106
↓ 2 callersClassSlimOrcaIDK
SlimOrcaIDK data module for supervised finetuning.
lit_gpt/datamodules/slim_orca_idontknow.py:36
↓ 2 callersClassStyle
lit_gpt/datamodules/base_test.py:15
↓ 2 callersClassincremental_save
lit_gpt/utils_old.py:365
↓ 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:125
↓ 1 callersClassBlock
lit_gpt/lora.py:600
↓ 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:113
↓ 1 callersClassBlock
lit_gpt/model.py:245
↓ 1 callersClassBpeVocab
tokenizer/convert/convert.py:337
↓ 1 callersClassCapybara
Capybara data module for supervised finetuning.
lit_gpt/datamodules/capybara.py:38
↓ 1 callersClassCausalSelfAttention
A modification of `lit_gpt.model.CausalSelfAttention` that adds the attention over the adaption prompt.
lit_gpt/adapter.py:167
↓ 1 callersClassCausalSelfAttention
lit_gpt/lora.py:612
↓ 1 callersClassCausalSelfAttention
lit_gpt/adapter_v2.py:129
↓ 1 callersClassCausalSelfAttention
lit_gpt/model.py:283
↓ 1 callersClassCodeLlama
lit_gpt/prompts.py:260
↓ 1 callersClassCombinedDatasetIterator
lit_gpt/packed_dataset.py:296
↓ 1 callersClassDefault
lit_gpt/prompts.py:36
↓ 1 callersClassErrorCountHandler
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 callersClassEvalArgs
Evaluation-related arguments
lit_gpt/args.py:59
↓ 1 callersClassFalcon
lit_gpt/prompts.py:160
↓ 1 callersClassFreeWilly2
lit_gpt/prompts.py:235
↓ 1 callersClassFusedCrossEntropyLoss
lit_gpt/fused_cross_entropy.py:140
↓ 1 callersClassGemma
lit_gpt/prompts.py:300
↓ 1 callersClassHfVocab
tokenizer/convert/convert.py:458
↓ 1 callersClassIncrementalPyTorchPickler
lit_gpt/utils_old.py:309
↓ 1 callersClassIncrementalPyTorchPickler
lit_gpt/utils.py:172
↓ 1 callersClassLazyLoadingUnpickler
lit_gpt/utils_old.py:194
↓ 1 callersClassLazyStorage
tokenizer/convert/convert.py:774
↓ 1 callersClassLazyUnpickler
tokenizer/convert/convert.py:780
↓ 1 callersClassLlama2
lit_gpt/prompts.py:221
↓ 1 callersClassLlama2FunctionCalling
lit_gpt/prompts.py:186
↓ 1 callersClassLlamaInstruct
LlamaInstruct data module for supervised finetuning.
lit_gpt/datamodules/llama_instruct.py:31
↓ 1 callersClassLoRAQKVLinear
lit_gpt/lora.py:157
↓ 1 callersClassMockTokenizer
A dummy tokenizer that encodes each character as its ASCII code.
conftest.py:24
↓ 1 callersClassMovingAverage
finetune/full.py:57
↓ 1 callersClassNectar
dpo/adapters/nectar.py:24
↓ 1 callersClassNotYetLoadedTensor
lit_gpt/utils_old.py:94
↓ 1 callersClassNousResearch
lit_gpt/prompts.py:250
↓ 1 callersClassPackedDataset
lit_gpt/packed_dataset.py:40
↓ 1 callersClassPackedDatasetIterator
lit_gpt/packed_dataset.py:179
↓ 1 callersClassPhi1
lit_gpt/prompts.py:269
↓ 1 callersClassPhi2
lit_gpt/prompts.py:284
↓ 1 callersClassPlatypus
lit_gpt/prompts.py:245
↓ 1 callersClassQ8_0QuantizedDataType
tokenizer/convert/convert.py:90
↓ 1 callersClassSavingProxyForStorage
lit_gpt/utils_old.py:267
↓ 1 callersClassSavingProxyForTensor
lit_gpt/utils_old.py:295
↓ 1 callersClassSavingProxyForTensor
lit_gpt/utils.py:140
↓ 1 callersClassSentencePieceVocab
tokenizer/convert/convert.py:395
↓ 1 callersClassSlimOrcaDedup
SlimOrcaDedup data module for supervised finetuning.
lit_gpt/datamodules/slim_orca_dedup.py:38
↓ 1 callersClassSpeedMonitorBase
Logs the training throughput and utilization. +-------------------------------------+-----------------------------------------------------------+
lit_gpt/speed_monitor.py:153
↓ 1 callersClassSplitter
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 callersClassStableCode
lit_gpt/prompts.py:255
↓ 1 callersClassStableLMAlpha
lit_gpt/prompts.py:103
↓ 1 callersClassStableLMZephyr
lit_gpt/prompts.py:122
↓ 1 callersClassStandardParquetPreparer
prepare_dataset/standard_parquet.py:17
↓ 1 callersClassTinyLlama
lit_gpt/prompts.py:289
↓ 1 callersClassTogetherComputerChat
lit_gpt/prompts.py:127
↓ 1 callersClassTogetherComputerInstruct
lit_gpt/prompts.py:141
↓ 1 callersClassTrainArgs
Training-related arguments
lit_gpt/args.py:8
↓ 1 callersClassUltraChat
UltraChat data module for supervised finetuning.
lit_gpt/datamodules/ultrachat_200k.py:38
↓ 1 callersClassVicuna
lit_gpt/prompts.py:177
↓ 1 callersClassVocabFactory
tokenizer/convert/convert.py:1287
↓ 1 callersClassWizardLMEvolInstructV2
WizardLMEvolInstructV2 data module for supervised finetuning.
lit_gpt/datamodules/wizardlm_evol_instruct_v2.py:40
↓ 1 callersClassincremental_save
lit_gpt/utils.py:228
ClassAlpaca
lit_gpt/prompts.py:70
ClassApplyRotaryEmb
lit_gpt/fused_rotary_embedding.py:10
ClassCapybaraMessage
lit_gpt/datamodules/capybara.py:27
ClassCapybaraRow
lit_gpt/datamodules/capybara.py:32
ClassCombinedDataset
lit_gpt/packed_dataset.py:283
ClassConfig
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
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:458
ClassConfig
lit_gpt/adapter_v2.py:27
ClassConfigBase
lit_gpt/config_base.py:12
ClassDataArguments
finetune/custom_dataclasses.py:16
ClassDataCollatorForCausalLM
finetune/datacollators.py:246
ClassDataCollatorForCausalLMDecoderOnlyChatML
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
ClassDataModule
Base class for all data modules in LitGPT.
lit_gpt/datamodules/base.py:16
ClassDataType
tokenizer/convert/convert.py:55
ClassDatasetAdapter
dpo/adapters/base.py:18
ClassDatasetPreparer
prepare_dataset/preparer.py:54
next →1–100 of 173, ranked by callers