MCPcopy Create free account

hub / github.com/allenai/OLMo / functions

Functions3,821 in github.com/allenai/OLMo

↓ 1 callersFunctioncompare_module_outputs
( base_traces_folder: Path, compare_traces_folder: Path, *, include_non_tensor_outputs: bool =
scripts/compare_module_outputs.py:102
↓ 1 callersMethodcompute
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/metrics/accuracy.py:26
↓ 1 callersMethodcompute
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/metrics/accuracy.py:26
↓ 1 callersMethodcompute_metrics
(self)
olmo/eval/evaluator.py:29
↓ 1 callersMethodcompute_scores
(gold_list, pred)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/coqa.py:102
↓ 1 callersMethodcompute_scores
(gold_list, pred)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/coqa.py:102
↓ 1 callersMethodconfigure
( self, bits, perchannel=False, sym=True, mse=False, norm=2.4, grid=100, maxshrink=0.8, trits=False
inference/compression/dependencies/AutoGPTQ/auto_gptq/quantization/quantizer.py:23
↓ 1 callersFunctionconvert_checkpoint
(checkpoint_dir: str, ignore_olmo_compatibility: bool = False)
hf_olmo/convert_olmo_to_hf.py:103
↓ 1 callersMethodcount_bytes
(cls, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:679
↓ 1 callersMethodcount_bytes
(cls, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:679
↓ 1 callersMethodcount_words
Downstream tasks with custom word boundaries should override this!
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:683
↓ 1 callersMethodcount_words
Downstream tasks with custom word boundaries should override this!
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:683
↓ 1 callersMethodcreate_from_arg_string
(cls, arg_string, additional_config=None)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:115
↓ 1 callersMethodcreate_from_arg_string
(cls, arg_string, additional_config=None)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:115
↓ 1 callersFunctioncreate_model_config_from_pretrained_config
Utility function
hf_olmo/modeling_olmo.py:19
↓ 1 callersFunctioncreate_task
(subject)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_test.py:96
↓ 1 callersFunctioncreate_task
(subject)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_test.py:96
↓ 1 callersMethodcreate_temp_file
(self, suffix: Optional[str] = None)
scripts/storage_cleaner.py:143
↓ 1 callersFunctioncreate_translation_task
(dataset, language_pair, version=0)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/translation.py:84
↓ 1 callersFunctioncreate_translation_task
(dataset, language_pair, version=0)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/translation.py:84
↓ 1 callersFunctioncross_entropy_loss
( logits, labels, ignore_index: int = -100, reduction: str = "mean", compute_z_loss: bool
olmo/train.py:123
↓ 1 callersMethoddataset
(self, split: str)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/mrqa.py:21
↓ 1 callersMethoddataset
(self, split: str)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/huggingface.py:69
↓ 1 callersMethoddataset
(self, split: str)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/mrqa.py:21
↓ 1 callersMethoddataset
(self, split: str)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/huggingface.py:69
↓ 1 callersFunctiondecode_key
(key: str)
olmo/safetensors_util.py:29
↓ 1 callersFunctiondelete_bad_runs
(run_paths: List[str], config: DeleteBadRunsConfig)
scripts/storage_cleaner.py:756
↓ 1 callersMethoddet_hash_object
Return an object to use for deterministic hashing instead of ``self``.
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tango_utils/det_hash.py:35
↓ 1 callersMethoddet_hash_object
Return an object to use for deterministic hashing instead of ``self``.
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tango_utils/det_hash.py:35
↓ 1 callersMethoddetokenize
(self, text)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/cbt.py:56
↓ 1 callersMethoddetokenize
(self, text)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/cbt.py:56
↓ 1 callersMethoddoc_to_continuations
Match EAI eval harness returns a list of continuations
olmo/eval/downstream.py:383
↓ 1 callersMethoddoc_to_continuations
(self, doc)
olmo/eval/downstream.py:599
↓ 1 callersMethoddoc_to_decontamination_query
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:453
↓ 1 callersMethoddoc_to_decontamination_query
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:453
↓ 1 callersMethoddoc_to_domain_conditional
Provide string for domain conditional normalization by default its blank string, continuation normalized by prob conditioned on a blank
olmo/eval/downstream.py:396
↓ 1 callersMethoddoc_to_domain_conditional
same number of domain conditionals as context
olmo/eval/downstream.py:607
↓ 1 callersMethoddoc_to_domain_conditional
(self, doc)
olmo/eval/downstream.py:1018
↓ 1 callersMethoddoc_to_label
Match EAI eval harness returns continuation id which corresponds to true label
olmo/eval/downstream.py:390
↓ 1 callersMethoddoc_to_label
(self, doc)
olmo/eval/downstream.py:604
↓ 1 callersMethoddoc_to_label
(self, doc)
olmo/eval/downstream.py:828
↓ 1 callersMethoddoc_to_target
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:462
↓ 1 callersMethoddoc_to_target
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:653
↓ 1 callersMethoddoc_to_target
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/lambada.py:57
↓ 1 callersMethoddoc_to_target
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/asdiv.py:71
↓ 1 callersMethoddoc_to_target
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:462
↓ 1 callersMethoddoc_to_target
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:653
↓ 1 callersMethoddoc_to_target
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/lambada.py:57
↓ 1 callersMethoddoc_to_target
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/asdiv.py:71
↓ 1 callersMethoddoc_to_text
Match EAI eval harness returns a single context string
olmo/eval/downstream.py:376
↓ 1 callersMethoddoc_to_text
(self, doc)
olmo/eval/downstream.py:589
↓ 1 callersMethoddownload
Downloads and returns the task dataset. Override this method to download the dataset from a custom API.
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:390
↓ 1 callersMethoddownload
Downloads and returns the task dataset. Override this method to download the dataset from a custom API.
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:390
↓ 1 callersMethoddownload_folder
(self, directory_path: str, local_dest_folder: PathOrStr)
scripts/storage_cleaner.py:208
↓ 1 callersFunctiondownload_s3_directory
(bucket_name: str, prefix: str, local_dir: str, ignore: str | None = None)
hf_olmo/convert_olmo_to_hf.py:146
↓ 1 callersMethoddummy_predict
(self, dummy_inputs: List[Dict[str, Any]], max_batch_size: int)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/stdio_wrapper.py:143
↓ 1 callersMethoddummy_predict
(self, dummy_inputs: List[Dict[str, Any]], max_batch_size: int)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/stdio_wrapper.py:143
↓ 1 callersFunctionduplicate_interleave
A simple version of `torch.repeat_interleave` for duplicating a matrix while interleaving the copy.
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/fused_gptj_attn.py:30
↓ 1 callersFunctionefficiency_benchmark_prompt_conversion
( **kwargs, )
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/efficiency_benchmark.py:241
↓ 1 callersFunctionefficiency_benchmark_prompt_conversion
( **kwargs, )
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/efficiency_benchmark.py:241
↓ 1 callersFunctionencode_fn
(row)
scripts/prepare_memmap_dataset.py:472
↓ 1 callersFunctionencode_key
(key: STKey)
olmo/safetensors_util.py:23
↓ 1 callersMethodeval_batch
(self, batch: Dict[str, Any])
olmo/train.py:902
↓ 1 callersFunctioneval_hf_model
(args, subject, model, tokenizer, dev_df, test_df, batch_size=1)
inference/eval/mmlu/run_eval.py:50
↓ 1 callersFunctioneval_openai_chat_engine
(args, subject, engine, dev_df, test_df, batch_size=1)
inference/eval/mmlu/run_eval.py:99
↓ 1 callersMethodeval_step
(self, batch: Dict[str, Any], evaluator: Evaluator)
olmo/train.py:907
↓ 1 callersFunctionevaluate
Instantiate and evaluate a model on a list of tasks. :param lm: obj Language Model :param task_dict: dict[str, Task] Dictiona
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/evaluator.py:115
↓ 1 callersFunctionevaluate
Instantiate and evaluate a model on a list of tasks. :param lm: obj Language Model :param task_dict: dict[str, Task] Dictiona
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/evaluator.py:115
↓ 1 callersFunctionext_make_q4
Construct Q4Matrix, return handle
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/qlinear/qlinear_exllama.py:23
↓ 1 callersFunctionext_q4_matmul
Matrix multiplication, returns x @ q4
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/qlinear/qlinear_exllama.py:28
↓ 1 callersFunctionf1_score
(items)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/metrics.py:39
↓ 1 callersMethodfasterquant
(self, blocksize=128, percdamp=0.01, group_size=-1, actorder=False, static_groups=False)
inference/compression/dependencies/AutoGPTQ/auto_gptq/quantization/gptq.py:61
↓ 1 callersMethodfewshot_context
(self, doc, num_fewshot, provide_description=None, rnd=None, description=None)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/truthfulqa.py:204
↓ 1 callersMethodfewshot_context
(self, doc, num_fewshot, provide_description=None, rnd=None, description=None)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/truthfulqa.py:204
↓ 1 callersMethodfewshot_examples
(self, k, rnd)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:447
↓ 1 callersMethodfewshot_examples
(self, k, rnd)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:447
↓ 1 callersFunctionfile_size
Get the size of a local or remote file in bytes.
olmo/util.py:331
↓ 1 callersFunctionfilter_warnings
()
olmo/util.py:175
↓ 1 callersFunctionfind_all_linear_names
( model: BaseGPTQForCausalLM, ignore: Optional[List[str]] = None, ignore_lm_head: bool = True )
inference/compression/dependencies/AutoGPTQ/auto_gptq/utils/peft_utils.py:311
↓ 1 callersFunctionfind_periodic_sequences
Function to find periodic sequences in an array. This function sweeps through the array and checks for sequences of length [min_period, max_p
olmo/data/util.py:41
↓ 1 callersFunctionfind_start_last_consecutive_true
Function to find the start position of the last consecutive sequence of True in an array.
olmo/data/util.py:21
↓ 1 callersFunctionfind_test_root
Search upward in the directory tree to a maximum of three layers to find and return the package root (containing the 'tests' folder)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/utils.py:169
↓ 1 callersFunctionfind_test_root
Search upward in the directory tree to a maximum of three layers to find and return the package root (containing the 'tests' folder)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/utils.py:169
↓ 1 callersMethodfinish
(self, metadata: Metadata, results: List[List[WriteResult]])
olmo/checkpoint.py:400
↓ 1 callersMethodfit
(self)
olmo/train.py:1097
↓ 1 callersMethodfix_a_slash_b
(self, string)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_math.py:177
↓ 1 callersMethodfix_a_slash_b
(self, string)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_math.py:177
↓ 1 callersMethodfix_fracs
(self, string)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_math.py:146
↓ 1 callersMethodfix_fracs
(self, string)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_math.py:146
↓ 1 callersFunctionfix_optim_state_dict
Make sure old optim state dicts are compatible with new versions.
olmo/optim.py:911
↓ 1 callersMethodfix_sqrt
(self, string)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_math.py:200
↓ 1 callersMethodfix_sqrt
(self, string)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_math.py:200
↓ 1 callersFunctionfix_tokenizer
(checkpoint_dir: str, tokenizer_name_or_path: Optional[str] = None)
hf_olmo/convert_olmo_to_hf.py:118
↓ 1 callersFunctionfix_url_dependencies
Pip and setuptools disagree about how URL dependencies should be handled.
inference/efficiency/dependencies/efficiency-pentathlon/setup.py:9
↓ 1 callersFunctionflatten_dict
(d: Dict)
olmo/safetensors_util.py:35
↓ 1 callersFunctionformat_subject
(subject)
inference/eval/mmlu/run_eval.py:19
↓ 1 callersMethodfree
(self)
inference/compression/dependencies/AutoGPTQ/auto_gptq/quantization/gptq.py:188
↓ 1 callersMethodfrom_train_config
(cls, config: TrainConfig)
olmo/data/collator.py:20
↓ 1 callersFunctionfused_loss_fn
( logits, labels, ignore_index: int = -100, reduction: str = "mean", c
olmo/train.py:155
↓ 1 callersFunctiongen_model
(n, m, t, bits, p, gs)
inference/compression/dependencies/AutoGPTQ/autogptq_extension/qigen/generate.py:696
← previousnext →701–800 of 3,821, ranked by callers