MCPcopy Create free account

hub / github.com/OpenBitSys/BitDistiller / types & classes

Types & classes490 in github.com/OpenBitSys/BitDistiller

⨍ Functions 1,634◇ Types & classes 490
↓ 5 callersClassEthicsConfig
BuilderConfig for Hendrycks ETHICS.
test/general/lm_eval/datasets/hendrycks_ethics/hendrycks_ethics.py:50
↓ 4 callersClassSteN2F3Quantizer
quantization/quantizer.py:184
↓ 3 callersClassLPLayerNorm
inference/models/mpt.py:59
↓ 3 callersClassRMSNorm
inference/models/llama.py:23
↓ 3 callersClassRequest
test/general/lm_eval/base.py:952
↓ 2 callersClassCacheHook
test/general/lm_eval/base.py:866
↓ 2 callersClassQuantLlamaRotaryEmbedding
inference/modules/fused_attn.py:22
↓ 2 callersClassSteInt2AsymQuantizer
quantization/quantizer.py:153
↓ 2 callersClassSupervisedDataset
Dataset for supervised fine-tuning.
train/train.py:175
↓ 2 callersClassWQLinear
quantization/qmodule.py:41
↓ 2 callersClasseach
test/general/lm_eval/tasks/race.py:29
↓ 1 callersClassCatcher
quantization/pre_quant.py:106
↓ 1 callersClassCatcher
quantization/autoclip.py:132
↓ 1 callersClassCatcher
test/general/wiki_ppl.py:63
↓ 1 callersClassDataCollatorForSupervisedDataset
Collate examples for supervised fine-tuning.
data/generation/single_generate.py:126
↓ 1 callersClassDataCollatorForSupervisedDataset
Collate examples for supervised fine-tuning.
data/generation/generate.py:126
↓ 1 callersClassDataCollatorForSupervisedDataset
Collate examples for supervised fine-tuning.
test/gsm8k/test.py:169
↓ 1 callersClassDataCollatorForSupervisedDataset
Collate examples for supervised fine-tuning.
train/train.py:215
↓ 1 callersClassExitCodeError
test/general/lm_eval/utils.py:17
↓ 1 callersClassFTLlamaRMSNorm
inference/modules/fused_norm.py:7
↓ 1 callersClassFalconAttentionFused
inference/models/falcon.py:78
↓ 1 callersClassFalconMLP
inference/models/falcon.py:211
↓ 1 callersClassFalconSimplePrompter
inference/utils/prompt_templates.py:164
↓ 1 callersClassJanitor
test/general/lm_eval/decontamination/janitor.py:106
↓ 1 callersClassKDTrainer
train/mytrainer.py:26
↓ 1 callersClassLMEvalAdaptor
test/general/utils_eval.py:7
↓ 1 callersClassLlama2Prompter
inference/utils/prompt_templates.py:125
↓ 1 callersClassLlamaAttentionFused
inference/models/llama.py:73
↓ 1 callersClassLlamaMLP
inference/models/llama.py:224
↓ 1 callersClassMPTAttentionFused
inference/models/mpt.py:95
↓ 1 callersClassMPTBlock
inference/models/mpt.py:234
↓ 1 callersClassMPTChatPrompter
inference/utils/prompt_templates.py:201
↓ 1 callersClassMPTMLP
inference/models/mpt.py:222
↓ 1 callersClassMPTPrompter
inference/utils/prompt_templates.py:191
↓ 1 callersClassMetaMathPrompter
inference/utils/prompt_templates.py:158
↓ 1 callersClassMultiTokenEOSCriteria
Criteria to stop on the specified multi-token sequence.
test/general/lm_eval/models/huggingface.py:723
↓ 1 callersClassQLinear
quantization/qlinear.py:50
↓ 1 callersClassQuantLlamaAttention
Multi-headed attention from 'Attention Is All You Need' paper
inference/modules/fused_attn.py:79
↓ 1 callersClassQuantLlamaAttentionFused
inference/modules/fused_attn.py:166
↓ 1 callersClassQuantLlamaMLP
inference/modules/fused_mlp.py:18
↓ 1 callersClassRequestFactory
test/general/lm_eval/base.py:985
↓ 1 callersClassRotaryEmbedding
Implementation of RotaryEmbedding from GPT-NeoX. This implementation is design to operate on queries and keys that are compatible with [batch_
inference/models/falcon.py:27
↓ 1 callersClassSharedEmbedding
inference/models/mpt.py:88
↓ 1 callersClassSteInt3AsymQuantizer
quantization/quantizer.py:119
↓ 1 callersClassSupervisedDataset
Dataset for supervised fine-tuning.
data/generation/single_generate.py:93
↓ 1 callersClassSupervisedDataset
Dataset for supervised fine-tuning.
data/generation/generate.py:93
↓ 1 callersClassSupervisedDataset
Dataset for supervised fine-tuning.
test/gsm8k/test.py:106
↓ 1 callersClassTextReader
test/general/lm_eval/decontamination/archiver.py:94
↓ 1 callersClassTransformer
inference/models/falcon.py:254
↓ 1 callersClassTransformer
inference/models/llama.py:264
↓ 1 callersClassTransformer
inference/models/mpt.py:257
↓ 1 callersClassTransformerBlock
inference/models/falcon.py:227
↓ 1 callersClassTransformerBlock
inference/models/llama.py:238
↓ 1 callersClassVicunaPrompter
inference/utils/prompt_templates.py:115
↓ 1 callersClassZStdTextReader
test/general/lm_eval/decontamination/archiver.py:151
↓ 1 callersClass_bootstrap_internal
test/general/lm_eval/metrics.py:192
ClassANLIBase
test/general/lm_eval/tasks/anli.py:33
ClassANLIRound1
test/general/lm_eval/tasks/anli.py:133
ClassANLIRound2
test/general/lm_eval/tasks/anli.py:137
ClassANLIRound3
test/general/lm_eval/tasks/anli.py:141
ClassARCChallenge
test/general/lm_eval/tasks/arc.py:77
ClassARCEasy
test/general/lm_eval/tasks/arc.py:29
ClassASDiv
ASDiv: A Diverse Corpus for Evaluating and Developing English Math Word Problem Solvers
test/general/lm_eval/datasets/asdiv/asdiv.py:52
ClassAnagrams1
test/general/lm_eval/tasks/unscramble.py:77
ClassAnagrams2
test/general/lm_eval/tasks/unscramble.py:81
ClassAnthropicLM
test/general/lm_eval/models/anthropic_llms.py:38
ClassArchive
test/general/lm_eval/decontamination/archiver.py:21
ClassArithmetic
test/general/lm_eval/tasks/arithmetic.py:29
ClassArithmetic1DComposite
test/general/lm_eval/tasks/arithmetic.py:116
ClassArithmetic2DMinus
test/general/lm_eval/tasks/arithmetic.py:84
ClassArithmetic2DMultiplication
test/general/lm_eval/tasks/arithmetic.py:112
ClassArithmetic2DPlus
test/general/lm_eval/tasks/arithmetic.py:80
ClassArithmetic3DMinus
test/general/lm_eval/tasks/arithmetic.py:92
ClassArithmetic3DPlus
test/general/lm_eval/tasks/arithmetic.py:88
ClassArithmetic4DMinus
test/general/lm_eval/tasks/arithmetic.py:100
ClassArithmetic4DPlus
test/general/lm_eval/tasks/arithmetic.py:96
ClassArithmetic5DMinus
test/general/lm_eval/tasks/arithmetic.py:108
ClassArithmetic5DPlus
test/general/lm_eval/tasks/arithmetic.py:104
ClassAsdiv
test/general/lm_eval/tasks/asdiv.py:35
ClassAutoCausalLM
Causal language modeling. You can find a set of supported models in the HF documentation: https://huggingface.co/docs/transformers/main/model_
test/general/lm_eval/models/huggingface.py:497
ClassAutoSeq2SeqLM
Seq2Seq language modeling. You can find a set of supported models in the following documentation: https://huggingface.co/docs/transformers/mai
test/general/lm_eval/models/huggingface.py:564
ClassBabi
test/general/lm_eval/tasks/babi.py:19
ClassBaseLM
test/general/lm_eval/base.py:121
ClassBasePrompter
inference/utils/prompt_templates.py:11
ClassBigBenchJsonTask
test/general/lm_eval/tasks/bigbench.py:32
ClassBlimpAdjunctIsland
test/general/lm_eval/tasks/blimp.py:118
ClassBlimpAnaphorGenderAgreement
test/general/lm_eval/tasks/blimp.py:122
ClassBlimpAnaphorNumberAgreement
test/general/lm_eval/tasks/blimp.py:126
ClassBlimpAnimateSubjectPassive
test/general/lm_eval/tasks/blimp.py:130
ClassBlimpAnimateSubjectTrans
test/general/lm_eval/tasks/blimp.py:134
ClassBlimpCausative
test/general/lm_eval/tasks/blimp.py:138
ClassBlimpComplex_NPIsland
test/general/lm_eval/tasks/blimp.py:142
ClassBlimpCoordinateStructureConstraintComplexLeftBranch
test/general/lm_eval/tasks/blimp.py:146
ClassBlimpCoordinateStructureConstraintObjectExtraction
test/general/lm_eval/tasks/blimp.py:150
ClassBlimpDeterminerNounAgreementIrregular_1
test/general/lm_eval/tasks/blimp.py:162
ClassBlimpDeterminerNounAgreementIrregular_2
test/general/lm_eval/tasks/blimp.py:166
ClassBlimpDeterminerNounAgreementWithAdjIrregular_1
test/general/lm_eval/tasks/blimp.py:174
ClassBlimpDeterminerNounAgreementWithAdjIrregular_2
test/general/lm_eval/tasks/blimp.py:178
ClassBlimpDeterminerNounAgreementWithAdj_2
test/general/lm_eval/tasks/blimp.py:170
ClassBlimpDeterminerNounAgreementWithAdjective_1
test/general/lm_eval/tasks/blimp.py:182
next →1–100 of 490, ranked by callers