MCPcopy Create free account

hub / github.com/OpenSparseLLMs/MoM / functions

Functions861 in github.com/OpenSparseLLMs/MoM

↓ 12 callersMethodadd_partial
(self, attr, req, res)
lm-eval-harness/lm_eval/api/model.py:168
↓ 10 callersMethodapply
Defines the operation to perform on a list of the `inst.resps` properties of `Instance` objects. Should return the list of (filtered)
lm-eval-harness/lm_eval/api/filter.py:23
↓ 9 callersMethodmodel
(self)
lm-eval-harness/lm_eval/models/huggingface.py:361
↓ 8 callersMethoddevice
(self)
lm-eval-harness/lm_eval/models/huggingface.py:396
↓ 8 callersMethoddoc_to_target
(self, doc)
lm-eval-harness/lm_eval/api/task.py:348
↓ 8 callersMethoddoc_to_target
(self, doc: dict)
lm-eval-harness/lm_eval/api/task.py:1049
↓ 7 callersMethoddoc_to_choice
(self, doc: Any)
lm-eval-harness/lm_eval/api/task.py:1093
↓ 6 callersMethodgenerate
r""" A streamlined generate() method overriding the transformers.GenerationMixin.generate() method. This method uses the same logits
lm-eval-harness/lm_eval/models/neuron_optimum.py:87
↓ 6 callersMethodget_reordered
Gets the reordered array Returns: List[Any]: The reordered array
lm-eval-harness/lm_eval/utils.py:187
↓ 6 callersMethodload_state_dict
(self, state_dict)
training/flame/data.py:147
↓ 6 callersMethodtok_encode
(self, string: str)
lm-eval-harness/lm_eval/models/openai_completions.py:174
↓ 5 callersMethod_get_config
(self, name)
lm-eval-harness/lm_eval/tasks/__init__.py:111
↓ 5 callersFunction_process_doc_prepended_question
(doc)
lm-eval-harness/lm_eval/tasks/scrolls/task.py:61
↓ 5 callersMethoddoc_to_text
(self, doc)
lm-eval-harness/lm_eval/api/task.py:344
↓ 5 callersMethodhandle_recurrent_state
(self, recurrent_state, recurrent_state_new, cu_seqlens, cu_seqlen_all, reverse_indices)
mom/layers/mom.py:778
↓ 5 callersFunctionpad_and_concat
Method for padding a list of tensors given the maximum tensor length in the batch. Used for batching inputs and continuations in seq2seq
lm-eval-harness/lm_eval/models/utils.py:140
↓ 5 callersMethodtest_docs
:return: Iterable[obj] A iterable of any object, that doc_to_text can handle
lm-eval-harness/lm_eval/api/task.py:290
↓ 5 callersMethodtok_encode
lm-eval-harness/lm_eval/models/neuron_optimum.py:343
↓ 5 callersMethodtok_encode
lm-eval-harness/lm_eval/models/huggingface.py:676
↓ 5 callersMethodvalidation_docs
:return: Iterable[obj] A iterable of any object, that doc_to_text can handle
lm-eval-harness/lm_eval/api/task.py:283
↓ 4 callersMethod__init__
( self, hidden_size: int, hidden_ratio: Optional[int] = None, intermediate_siz
mom_naive/models/mom_linear_attn/modeling_mom_linear_attn.py:34
↓ 4 callersMethod__init__
( self, hidden_size: int, hidden_ratio: Optional[int] = None, intermediate_siz
mom_naive/models/mom_gated_deltanet/modeling_mom_gated_deltanet.py:39
↓ 4 callersMethod__init__
( self, hidden_size: int, hidden_ratio: Optional[int] = None, intermediate_siz
mom_naive/models/mom_gsa/modeling_mom_gsa.py:34
↓ 4 callersMethod__init__
( self, hidden_size: int, hidden_ratio: Optional[int] = None, intermediate_siz
mom_naive/models/mom_gla/modeling_mom_gla.py:33
↓ 4 callersMethod_name_is_registered
(self, name)
lm-eval-harness/lm_eval/tasks/__init__.py:72
↓ 4 callersMethoddoc_to_text
(self, doc)
lm-eval-harness/lm_eval/api/task.py:1015
↓ 4 callersMethodget_batched
Generates and yields batches from the reordered array. Parameters: - n (int): The size of each batch. Defaults to 1.
lm-eval-harness/lm_eval/models/utils.py:383
↓ 4 callersFunctionget_logger
(name: str = None)
training/flame/logging.py:14
↓ 4 callersFunctionget_metric
(name: str, hf_evaluate_metric=False)
lm-eval-harness/lm_eval/api/registry.py:121
↓ 4 callersFunctionget_metric_aggregation
(name: str)
lm-eval-harness/lm_eval/api/registry.py:158
↓ 4 callersMethodget_original
(self, grouped_dict)
lm-eval-harness/lm_eval/models/utils.py:118
↓ 4 callersMethodget_original
Restores the original order of elements from the reordered list. Parameters: - newarr (List): The reordered array.
lm-eval-harness/lm_eval/models/utils.py:421
↓ 4 callersMethodhas_validation_docs
Whether the task has a validation set
lm-eval-harness/lm_eval/api/task.py:267
↓ 4 callersFunctionis_higher_better
(metric_name)
lm-eval-harness/lm_eval/api/registry.py:165
↓ 4 callersFunctionis_non_str_iterable
(obj)
lm-eval-harness/lm_eval/api/metrics.py:333
↓ 4 callersFunctionmetric_max_over_ground_truths
Compute max metric between prediction and each ground truth.
lm-eval-harness/lm_eval/api/metrics.py:319
↓ 4 callersFunctionnormalize_squad
Normalization used in official SQuAD evaluation script.
lm-eval-harness/lm_eval/tasks/super_glue/record/t5_utils.py:50
↓ 4 callersMethodprepare_recurrent_state
(self, recurrent_state, cu_seqlens, cu_seqlen_all, reverse_indices, batch_size)
mom/layers/mom.py:756
↓ 4 callersMethodrandint
( self, low: int, high: int, batch_size: int = 1024, g: torch.Generato
training/flame/data.py:114
↓ 4 callersMethodsample
Draw the first `n` samples in order from the specified split. Used for tasks with "canonical" ordered fewshot examples, such as MMLU
lm-eval-harness/lm_eval/api/samplers.py:75
↓ 4 callersMethodtok_encode
lm-eval-harness/lm_eval/models/vllm_causallms.py:146
↓ 3 callersMethod__init__
(self, config: MomConfig, layer_idx: int)
mom/models/mom/modeling_mom.py:116
↓ 3 callersMethod_detect_batch_size
(self, requests=None, pos: int = 0)
lm-eval-harness/lm_eval/models/huggingface.py:621
↓ 3 callersMethod_process_doc
(self, doc)
lm-eval-harness/lm_eval/tasks/scrolls/task.py:451
↓ 3 callersFunction_sacreformat
Format refs and preds for sacrebleu corpus calculation. It is very particular
lm-eval-harness/lm_eval/api/metrics.py:337
↓ 3 callersFunctionbenchmark_backward
Use Pytorch Benchmark on the backward pass of an arbitrary function.
benchmarks/ops/benchmark.py:30
↓ 3 callersFunctionbenchmark_forward
Use Pytorch Benchmark on the forward pass of an arbitrary function.
benchmarks/ops/benchmark.py:8
↓ 3 callersFunctionbuild_filter_ensemble
Create a filtering pipeline.
lm-eval-harness/lm_eval/filters/__init__.py:33
↓ 3 callersFunctionconvert_choice
(choice)
lm-eval-harness/lm_eval/tasks/super_glue/copa/utils.py:1
↓ 3 callersMethoddoc_to_target
(self, doc)
lm-eval-harness/lm_eval/api/task.py:1444
↓ 3 callersMethodget_original
Restores the original order of a new array based on the old array's order Args: newarr (List[Any]): The array to be restored
lm-eval-harness/lm_eval/utils.py:195
↓ 3 callersMethodgroup
Groups elements of an iterable based on a provided function. Parameters: - arr (Iterable): The iterable to be grouped.
lm-eval-harness/lm_eval/models/utils.py:446
↓ 3 callersMethodhas_test_docs
Whether the task has a test set
lm-eval-harness/lm_eval/api/task.py:272
↓ 3 callersFunctionhash_args
(attr, args)
lm-eval-harness/lm_eval/api/model.py:155
↓ 3 callersFunctionmean
(arr)
lm-eval-harness/lm_eval/api/metrics.py:25
↓ 3 callersMethodnormalize_string
(self, s: str)
lm-eval-harness/lm_eval/decontamination/janitor.py:210
↓ 3 callersFunctionoa_completion
Query OpenAI API for completion. Retry with back-off until they respond
lm-eval-harness/lm_eval/models/openai_completions.py:45
↓ 3 callersFunctionreconstruct
Reconstruct and mix transformed outputs back into the original input sequence shape. Key operations: 1. Reshapes and transposes `transfo
mom_naive/layers/mom_gla.py:116
↓ 3 callersFunctionreconstruct
Reconstruct and mix transformed outputs back into the original input sequence shape. Key operations: 1. Reshapes and transposes `transfo
mom_naive/layers/mom_linear_attn.py:112
↓ 3 callersFunctionsizeof_fmt
(num, suffix='B')
benchmarks/benchmark_training_throughput.py:24
↓ 2 callersMethod_config_is_python_task
(self, config)
lm-eval-harness/lm_eval/tasks/__init__.py:102
↓ 2 callersMethod_config_is_task
(self, config)
lm-eval-harness/lm_eval/tasks/__init__.py:92
↓ 2 callersMethod_get_tasklist
(self, name)
lm-eval-harness/lm_eval/tasks/__init__.py:119
↓ 2 callersMethod_get_yaml_path
(self, name)
lm-eval-harness/lm_eval/tasks/__init__.py:107
↓ 2 callersMethod_loglikelihood_tokens
( self, requests: List[Tuple[Tuple[str, str], List[int], List[int]]], disable_tqdm: bo
lm-eval-harness/lm_eval/models/vllm_causallms.py:342
↓ 2 callersMethod_loglikelihood_tokens
( self, requests, disable_tqdm: bool = False, override_bs=None )
lm-eval-harness/lm_eval/models/neuron_optimum.py:531
↓ 2 callersMethod_loglikelihood_tokens
( self, requests, disable_tqdm: bool = False )
lm-eval-harness/lm_eval/models/openai_completions.py:209
↓ 2 callersMethod_loglikelihood_tokens
( self, requests: List[Tuple[Tuple[str, str], List[int], List[int]]], disable_tqdm: bo
lm-eval-harness/lm_eval/models/huggingface.py:954
↓ 2 callersMethod_model_call
:param inps: torch.Tensor A torch tensor of shape [batch, (sequence_ctx + sequence_cont)] or of shape [batch, sequenc
lm-eval-harness/lm_eval/models/huggingface.py:732
↓ 2 callersMethod_model_generate
( self, requests: List[List[int]] = None, generate: bool = False, max_tokens:
lm-eval-harness/lm_eval/models/vllm_causallms.py:164
↓ 2 callersMethod_model_generate
(self, context, max_length, stop, **generation_kwargs)
lm-eval-harness/lm_eval/models/huggingface.py:761
↓ 2 callersMethod_name_is_task
(self, name)
lm-eval-harness/lm_eval/tasks/__init__.py:77
↓ 2 callersMethod_normalize_answer
(text)
lm-eval-harness/lm_eval/tasks/scrolls/task.py:351
↓ 2 callersMethod_reorder
Reorders the elements in the array based on the sorting function. Parameters: - arr (Union[List, Tuple[Tuple[int, Any], ...]
lm-eval-harness/lm_eval/models/utils.py:407
↓ 2 callersMethod_scrolls_metrics
(self)
lm-eval-harness/lm_eval/tasks/scrolls/task.py:193
↓ 2 callersMethod_split_chunks
( self, dirty_string: str, dirty_parts: Sequence[Tuple] )
lm-eval-harness/lm_eval/decontamination/janitor.py:170
↓ 2 callersFunctionbenchmark_combined
Use Pytorch Benchmark on the forward+backward pass of an arbitrary function.
benchmarks/ops/benchmark.py:72
↓ 2 callersFunctionclean
Ignore capitalization and determiners.
lm-eval-harness/lm_eval/tasks/super_glue/wsc/t5_utils.py:79
↓ 2 callersMethodcleaned_context
(self, context)
lm-eval-harness/lm_eval/tasks/based_triviaqa/task.py:38
↓ 2 callersFunctionclear_torch_cache
()
lm-eval-harness/lm_eval/models/utils.py:191
↓ 2 callersFunctioncli_evaluate
(args: Union[argparse.Namespace, None] = None)
lm-eval-harness/lm_eval/__main__.py:234
↓ 2 callersMethodconstruct_requests
Uses RequestFactory to construct Requests and returns an iterable of Requests which will be sent to the LM. :param doc: T
lm-eval-harness/lm_eval/api/task.py:490
↓ 2 callersMethoddownload
Downloads and returns the task dataset. Override this method to download the dataset from a custom API. :param data_dir: str
lm-eval-harness/lm_eval/api/task.py:223
↓ 2 callersFunctionextended_json_dump
Extended JSON dump that can handle pydantic models.
lm-eval-harness/lm_eval/models/local_utils/jrt_utils.py:112
↓ 2 callersMethodfewshot_context
Returns a fewshot context string that is made up of a prepended description (if provided), the `num_fewshot` number of examples, and an append
lm-eval-harness/lm_eval/api/task.py:555
↓ 2 callersMethodfewshot_docs
(self)
lm-eval-harness/lm_eval/api/task.py:925
↓ 2 callersFunctionflatten_dict
Flatten a multi-level nested collection of dictionaries and lists into a flat dictionary. The function traverses nested dictionaries and
lm-eval-harness/lm_eval/models/local_utils/jrt_utils.py:51
↓ 2 callersFunctionform_ngrams
(sequence: Iterator[T], n: int)
lm-eval-harness/lm_eval/decontamination/janitor.py:24
↓ 2 callersMethodget_chunks
Divides an iterable into chunks of specified size or based on a given function. Useful for batching Parameters: - it
lm-eval-harness/lm_eval/models/utils.py:478
↓ 2 callersMethodget_config
(self, key: str)
lm-eval-harness/lm_eval/api/task.py:541
↓ 2 callersFunctionget_dtype
Converts `dtype` from `str` to torch.dtype when possible. Does not use an instantiated HF AutoConfig
lm-eval-harness/lm_eval/models/utils.py:196
↓ 2 callersMethodget_grouped
(self)
lm-eval-harness/lm_eval/models/utils.py:107
↓ 2 callersFunctionget_overlaps_dump_path
(task_name, task_set, ngrams_n_size, limit)
lm-eval-harness/lm_eval/decontamination/decontaminate.py:50
↓ 2 callersMethodgguf_completion
( self, context, continuation=None, stop=None, retries=3, delay=5, **kwargs )
lm-eval-harness/lm_eval/models/gguf.py:46
↓ 2 callersMethodhas_test_docs
(self)
lm-eval-harness/lm_eval/api/task.py:897
↓ 2 callersMethodhas_training_docs
Whether the task has a training set
lm-eval-harness/lm_eval/api/task.py:262
↓ 2 callersMethodhas_validation_docs
(self)
lm-eval-harness/lm_eval/api/task.py:891
↓ 2 callersFunctionload_config_hf
(model_name)
lm-eval-harness/lm_eval/models/jrt_lm.py:12
↓ 2 callersFunctionmake_table
Generate table of results.
lm-eval-harness/lm_eval/utils.py:217
↓ 2 callersFunctionprepare_inputs
( batch_size: int, seq_len: int, varlen: bool, vocab_size: int, device: torch.device )
benchmarks/benchmark_training_throughput.py:32
next →1–100 of 861, ranked by callers