Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/allenai/OLMo
/ functions
Functions
3,821 in github.com/allenai/OLMo
⨍
Functions
3,821
◇
Types & classes
799
↳
Endpoints
9
Method
_parse_labels
(self, label_ids: LongTensor)
inference/compression/dependencies/AutoGPTQ/auto_gptq/eval_tasks/language_modeling_task.py:37
Method
_parse_labels
(self, label_ids: LongTensor)
inference/compression/dependencies/AutoGPTQ/auto_gptq/eval_tasks/text_summarization_task.py:50
Method
_predict
(self, batch_data: Dict[str, Any], *args, **kwargs)
inference/compression/dependencies/AutoGPTQ/auto_gptq/eval_tasks/sequence_classification_task.py:58
Method
_predict
(self, batch_data: Dict[str, Any], *args, **kwargs)
inference/compression/dependencies/AutoGPTQ/auto_gptq/eval_tasks/language_modeling_task.py:31
Method
_predict
(self, batch_data: Dict[str, Any], *args, **kwargs)
inference/compression/dependencies/AutoGPTQ/auto_gptq/eval_tasks/text_summarization_task.py:33
Method
_prepare_summary
(summary)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/truthfulqa.py:380
Method
_prepare_summary
(summary)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/truthfulqa.py:380
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/drop.py:62
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/sciq.py:49
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/piqa.py:50
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/hellaswag.py:52
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/arc.py:53
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/wsc273.py:68
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/qa4mre.py:46
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_test.py:128
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/sat.py:60
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/headqa.py:52
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_ethics.py:353
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_math.py:51
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/openbookqa.py:54
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/logiqa.py:55
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/swag.py:49
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/prost.py:58
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/wikitext.py:87
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/quac.py:53
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/mathqa.py:52
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/drop.py:62
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/sciq.py:49
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/piqa.py:50
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/hellaswag.py:52
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/arc.py:53
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/wsc273.py:68
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/qa4mre.py:46
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_test.py:128
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/sat.py:60
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/headqa.py:52
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_ethics.py:353
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_math.py:51
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/openbookqa.py:54
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/logiqa.py:55
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/swag.py:49
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/prost.py:58
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/wikitext.py:87
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/quac.py:53
Method
_process_doc
(self, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/mathqa.py:52
Method
_rebuild_from_type_v2_monkey
(func, new_type, args, state)
olmo/checkpoint.py:1230
Function
_reduce8_acc
(a, b, c, d, e, f, g, h)
inference/compression/dependencies/AutoGPTQ/autogptq_extension/qigen/intrin.py:68
Function
_reduce_add
(a)
inference/compression/dependencies/AutoGPTQ/autogptq_extension/qigen/intrin.py:148
Function
_run_local_sharded_checkpointer_test
(rank: int, world_size: int, tmp_path: Path)
tests/checkpoint_test.py:28
Function
_run_local_unsharded_checkpointer
(rank: int, world_size: int, tmp_path: Path)
tests/ddp_checkpoint_test.py:31
Method
_scaled_dot_product_attention
( self, q: torch.Tensor, k: torch.Tensor, v: torch.Tensor, attn_mask:
olmo/model.py:882
Method
_shape
(self, tensor, seq_len, bsz)
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/fused_llama_attn.py:40
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/datasets/triviaqa/triviaqa.py:92
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/datasets/quac/quac.py:77
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/datasets/drop/drop.py:117
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/datasets/lambada_openai/lambada_openai.py:114
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/datasets/logiqa/logiqa.py:76
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/datasets/sat_analogies/sat_analogies.py:85
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/datasets/hendrycks_math/hendrycks_math.py:83
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/datasets/coqa/coqa.py:156
Method
_split_generators
Returns SplitGenerators.
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/datasets/wikitext/wikitext.py:114
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/datasets/mutual/mutual.py:77
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/datasets/hendrycks_ethics/hendrycks_ethics.py:144
Method
_split_generators
Returns SplitGenerators.
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/datasets/headqa/headqa.py:97
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/datasets/arithmetic/arithmetic.py:188
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/datasets/unscramble/unscramble.py:85
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/datasets/pile/pile.py:98
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/datasets/asdiv/asdiv.py:82
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/datasets/triviaqa/triviaqa.py:92
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/datasets/quac/quac.py:77
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/datasets/drop/drop.py:117
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/datasets/lambada_openai/lambada_openai.py:114
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/datasets/logiqa/logiqa.py:76
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/datasets/sat_analogies/sat_analogies.py:85
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/datasets/hendrycks_math/hendrycks_math.py:83
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/datasets/coqa/coqa.py:156
Method
_split_generators
Returns SplitGenerators.
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/datasets/wikitext/wikitext.py:114
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/datasets/mutual/mutual.py:77
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/datasets/hendrycks_ethics/hendrycks_ethics.py:144
Method
_split_generators
Returns SplitGenerators.
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/datasets/headqa/headqa.py:97
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/datasets/arithmetic/arithmetic.py:188
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/datasets/unscramble/unscramble.py:85
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/datasets/pile/pile.py:98
Method
_split_generators
(self, dl_manager)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/datasets/asdiv/asdiv.py:82
Function
_squad_agg
(key, items)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/squad.py:40
Function
_squad_agg
(key, items)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/squad.py:40
Function
_step_function_with_timestep
( last_predictions: torch.Tensor, state: dict[str, torch.Tensor], timestep: int, )
tests/beam_search_test.py:87
Method
_update_state
( self, state: ConstraintStateType, last_prediction: torch.Tensor, )
olmo/beam_search.py:624
Method
_wrap_up
()
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/efficiency/power_monitor.py:46
Method
_wrap_up
()
inference/efficiency/dependencies/previous_version/efficiency_benchmark/efficiency/power_monitor.py:46
Function
_wrapper
(*args, **kwargs)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/utils.py:156
Function
_wrapper
(*args, **kwargs)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/utils.py:156
Function
acc_all
(items)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/metrics.py:48
Function
acc_all
(items)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/metrics.py:48
Function
acc_all_stderr
(items)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/metrics.py:67
Function
acc_all_stderr
(items)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/metrics.py:67
Method
add_batch
(self, inp, out)
inference/compression/dependencies/AutoGPTQ/auto_gptq/quantization/gptq.py:33
Method
add_data
(self, data, meta={})
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/decontamination/archiver.py:32
Method
add_data
(self, data)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/decontamination/archiver.py:84
Method
add_data
(self, data, meta={})
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/decontamination/archiver.py:32
← previous
next →
1,501–1,600 of 3,821, ranked by callers