MCPcopy Create free account

hub / github.com/allenai/OLMo / functions

Functions3,821 in github.com/allenai/OLMo

Methodloglikelihood
(self, requests)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/models/textsynth.py:89
Methodloglikelihood
(self, requests)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:177
Methodloglikelihood
(self, requests)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/models/textsynth.py:89
Methodloglikelihood_rolling
(self, requests)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:192
Methodloglikelihood_rolling
(self, requests)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/models/textsynth.py:107
Methodloglikelihood_rolling
(self, requests)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/models/dummy.py:31
Methodloglikelihood_rolling
(self, requests)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:192
Methodloglikelihood_rolling
(self, requests)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/models/textsynth.py:107
Methodloglikelihood_rolling
(self, requests)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/models/dummy.py:31
Functionlorem_ipsum
()
conftest.py:87
Functionlorem_ipsum_docs
()
conftest.py:92
Functionmake_data_block
A simple implementation of text generation oriented smart batching to maximize VRAM usage when evaluation :param samples: Dict[str, List[str]], s
inference/compression/dependencies/AutoGPTQ/auto_gptq/utils/data_utils.py:13
Functionmake_disjoint_window
Takes output from get_rolling_token_windows and makes the context not overlap with the continuation
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/utils.py:116
Functionmake_disjoint_window
Takes output from get_rolling_token_windows and makes the context not overlap with the continuation
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/utils.py:116
Functionmake_table
Generate table of results.
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/evaluator.py:282
Functionmake_table
Generate table of results.
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/evaluator.py:282
Methodmanual_download_instructions
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/datasets/sat_analogies/sat_analogies.py:59
Methodmanual_download_instructions
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/datasets/sat_analogies/sat_analogies.py:59
Methodmassage_kwargs
(cls, kwargs: Dict[str, Any])
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/steps.py:102
Methodmassage_kwargs
(cls, kwargs: Dict[str, Any])
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/steps.py:211
Methodmassage_kwargs
(cls, kwargs: Dict[str, Any])
inference/efficiency/dependencies/previous_version/efficiency_benchmark/steps.py:98
Methodmassage_kwargs
(cls, kwargs: Dict[str, Any])
inference/efficiency/dependencies/previous_version/efficiency_benchmark/steps.py:203
Functionmatmul248_kernel_config_pruner
The main purpose of this function is to shrink BLOCK_SIZE_* when the corresponding dimension is smaller.
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/triton_utils/custom_autotune.py:153
Functionmatthews_corrcoef
(items)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/metrics.py:32
Functionmatthews_corrcoef
(items)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/metrics.py:32
Methodmax_epochs
(self)
olmo/train.py:257
Methodmax_gen_toks
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:138
Methodmax_gen_toks
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/models/gpt2.py:88
Methodmax_gen_toks
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/models/textsynth.py:68
Methodmax_gen_toks
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/models/gpt3.py:97
Methodmax_gen_toks
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:138
Methodmax_gen_toks
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/models/gpt2.py:88
Methodmax_gen_toks
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/models/textsynth.py:68
Methodmax_gen_toks
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/models/gpt3.py:97
Methodmax_length
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:133
Methodmax_length
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/models/gpt2.py:80
Methodmax_length
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/models/textsynth.py:63
Methodmax_length
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/models/gpt3.py:92
Methodmax_length
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:133
Methodmax_length
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/models/gpt2.py:80
Methodmax_length
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/models/textsynth.py:63
Methodmax_length
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/models/gpt3.py:92
Methodmax_seq_len
(self)
olmo/data/memmap_dataset.py:97
Methodmax_steps
(self)
olmo/train.py:261
Methodmax_tokens
(self)
olmo/train.py:281
Methodmc1
(lls)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/truthfulqa.py:126
Methodmc1
(lls)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/truthfulqa.py:126
Methodmc2
(lls)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/truthfulqa.py:130
Methodmc2
(lls)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/truthfulqa.py:130
Functionmedian
(arr)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/metrics.py:28
Functionmedian
(arr)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/metrics.py:28
Functionmemoize
(user_function, /)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/task.py:43
Functionmemoize
(user_function, /)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/task.py:43
Methodmerge
(self)
inference/compression/dependencies/AutoGPTQ/auto_gptq/utils/peft_utils.py:57
Methodmerge_adapter
(self)
inference/compression/dependencies/AutoGPTQ/auto_gptq/utils/peft_utils.py:168
Methodmerge_adapter
(self)
inference/compression/dependencies/AutoGPTQ/auto_gptq/utils/peft_utils.py:301
Methodmerge_and_unload
(self)
inference/compression/dependencies/AutoGPTQ/auto_gptq/utils/peft_utils.py:174
Methodmerge_and_unload
(self)
inference/compression/dependencies/AutoGPTQ/auto_gptq/utils/peft_utils.py:307
Functionmodel_config
()
conftest.py:44
Functionmodel_path
()
conftest.py:97
Functionmodule
(bits_list=[4, 2])
inference/compression/dependencies/AutoGPTQ/autogptq_extension/qigen/template.py:15
Methodnew_step
(last_predictions: torch.Tensor, state: Dict[str, torch.Tensor], time_step: int)
olmo/beam_search.py:805
Functionnodecounts_cmd
(args: argparse.Namespace)
scripts/ladder.py:313
Methodnormalize
(text)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/datasets/logiqa/logiqa.py:109
Methodnormalize
(text)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/datasets/logiqa/logiqa.py:109
Methodnum_attention_heads
(self)
hf_olmo/configuration_olmo.py:29
Methodnum_bck_flops
(self)
olmo/model.py:1587
Methodnum_fwd_flops
(self)
olmo/model.py:1569
Methodnum_hidden_layers
(self)
hf_olmo/configuration_olmo.py:33
Methodoffsets
(self)
olmo/data/memmap_dataset.py:102
Methodon_trace_ready
(p)
olmo/train.py:1145
Methodoutput_multiplier
(self)
olmo/model.py:334
Methodoutput_multiplier
(self)
olmo/model.py:351
Methodoutput_multiplier
(self)
olmo/model.py:357
Methodoutput_multiplier
(self)
olmo/model.py:367
Methodpack
(self, linear, scales, zeros, g_idx)
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/qlinear/qlinear_cuda_old.py:98
Methodpack
(self, linear, scales, zeros, g_idx=None)
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/qlinear/qlinear_triton.py:66
Methodpack
(self, linear, scales, zeros, g_idx=None)
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/qlinear/qlinear_exllama.py:100
Methodpad_token
(self)
olmo/tokenizer.py:54
Functionpartial_size
(s3_uri: str)
scripts/microanneal_config_maker.py:39
Functionpartial_size
(s3_uri: str)
scripts/peteish7_config_maker.py:38
Functionpass_through_fn
(fn, *args, **kwargs)
olmo/util.py:796
Functionpatched_get_worker_info
()
tests/data/iterable_dataset_test.py:89
Methodpath_choose
(*paths)
olmo/config.py:80
Methodpath_glob
(*paths)
olmo/config.py:70
Methodpath_last_checkpoint
(path)
olmo/config.py:92
Functionperplexity
(items)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/metrics.py:94
Functionperplexity
(items)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/metrics.py:94
Methodpersistent_id
(self, obj: Any)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tango_utils/det_hash.py:120
Methodpersistent_id
(self, obj: Any)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tango_utils/det_hash.py:120
Functionpop_stddev
(arr)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/metrics.py:14
Functionpop_stddev
(arr)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/metrics.py:14
Functionpositional_deprecated
A decorator to nudge users into passing only keyword args (`kwargs`) to the wrapped function, `fn`.
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/utils.py:149
Functionpositional_deprecated
A decorator to nudge users into passing only keyword args (`kwargs`) to the wrapped function, `fn`.
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/utils.py:149
Methodpost_init
(self)
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/qlinear/qlinear_cuda_old.py:95
Methodpost_init
(self)
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/qlinear/qlinear_triton.py:63
Methodpost_init
(self)
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/qlinear/qlinear_exllama.py:85
Methodpredict
(self, inputs)
inference/efficiency/run_efficiency_benchmark.py:32
Methodpredict
(self, inputs)
inference/efficiency/dependencies/previous_version/olmo_efficiency.py:71
Methodprep_examples
Overwrite for WinoGrande as multiple ctx, single continuation
olmo/eval/downstream.py:539
← previousnext →3,001–3,100 of 3,821, ranked by callers