MCPcopy Create free account

hub / github.com/allenai/OLMo / functions

Functions3,821 in github.com/allenai/OLMo

⨍ Functions 3,821◇ Types & classes 799↳ Endpoints 9
Functiontest_concat_mmap_datasets
(tmp_path: Path)
tests/data/memmap_dataset_test.py:84
Functiontest_config_save
(model_path: str)
tests/hf_olmo/configuration_olmo_test.py:6
Methodtest_cuda_old
(self, use_half2: bool)
inference/compression/dependencies/AutoGPTQ/tests/test_q4.py:1736
Methodtest_data_config
(self)
tests/config_test.py:55
Methodtest_different_per_node_beam_size
(self)
tests/beam_search_test.py:280
Functiontest_dir_is_empty
(tmp_path)
tests/util_test.py:4
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/sciq.py:46
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/cbt.py:53
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/triviaqa.py:51
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/arc.py:50
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/truthfulqa.py:77
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/truthfulqa.py:195
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/qa4mre.py:42
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/gsm8k.py:60
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/glue.py:178
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/glue.py:219
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/glue.py:506
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/webqs.py:52
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_test.py:125
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/pile.py:43
Methodtest_docs
:return: Iterable[obj] A iterable of any object, that doc_to_text can handle
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/translation.py:127
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/sat.py:57
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/storycloze.py:63
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/headqa.py:49
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_ethics.py:60
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_ethics.py:308
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/arithmetic.py:50
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_math.py:48
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/mutual.py:49
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/pubmedqa.py:47
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/lambada.py:44
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/openbookqa.py:51
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/asdiv.py:54
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/logiqa.py:52
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/anli.py:57
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/coqa.py:51
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/prost.py:51
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/wikitext.py:84
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/mc_taco.py:54
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/quac.py:50
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/mathqa.py:49
Methodtest_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/race.py:91
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/sciq.py:46
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/cbt.py:53
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/triviaqa.py:51
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/arc.py:50
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/truthfulqa.py:77
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/truthfulqa.py:195
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/qa4mre.py:42
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/gsm8k.py:60
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/glue.py:178
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/glue.py:219
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/glue.py:506
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/webqs.py:52
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_test.py:125
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/pile.py:43
Methodtest_docs
:return: Iterable[obj] A iterable of any object, that doc_to_text can handle
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/translation.py:127
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/sat.py:57
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/storycloze.py:63
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/headqa.py:49
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_ethics.py:60
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_ethics.py:308
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/arithmetic.py:50
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_math.py:48
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/mutual.py:49
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/pubmedqa.py:47
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/lambada.py:44
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/openbookqa.py:51
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/asdiv.py:54
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/logiqa.py:52
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/anli.py:57
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/coqa.py:51
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/prost.py:51
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/wikitext.py:84
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/mc_taco.py:54
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/quac.py:50
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/mathqa.py:49
Methodtest_docs
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/race.py:91
Methodtest_early_stopping
Checks case where beam search will reach `max_steps` before finding end tokens.
tests/beam_search_test.py:204
Methodtest_empty_sequences
(self)
tests/beam_search_test.py:308
Functiontest_encode
(tokenizer: Tokenizer, lorem_ipsum: str, add_special_tokens: bool)
tests/tokenizer_test.py:9
Functiontest_encode_batch
(tokenizer: Tokenizer, lorem_ipsum_docs: List[str], add_special_tokens: bool)
tests/tokenizer_test.py:28
Methodtest_exllama
(self)
inference/compression/dependencies/AutoGPTQ/tests/test_q4.py:1050
Methodtest_exllama_buffer_size
(self)
inference/compression/dependencies/AutoGPTQ/tests/test_q4.py:1106
Methodtest_finished_state
(self)
tests/beam_search_test.py:149
Functiontest_flash_attention_2
(model_path: str)
tests/hf_olmo/modeling_olmo_test.py:29
Functiontest_flash_attn
( train_config: TrainConfig, tokenizer: Tokenizer, positional_embeddings: Optional[str], block
tests/model_test.py:206
Functiontest_flatten_dict
()
tests/util_test.py:17
Functiontest_forward
( train_config: TrainConfig, tokenizer: Tokenizer, alibi: bool, rope: bool, flash_attn: bo
tests/model_test.py:115
Functiontest_forward
( train_config: TrainConfig, tokenizer: Tokenizer, alibi: bool, rope: bool, flash_attn: bo
tests/hf_olmo/hf_olmo_test.py:126
Functiontest_fused_loss
(batch_size, seq_len, vocab_size, z_loss_multiplier)
tests/train_test.py:14
Functiontest_generate
( train_config: TrainConfig, tokenizer: Tokenizer, cuda: bool, dtype: torch.dtype, )
tests/model_test.py:393
Methodtest_generation_no_act_order
(self)
inference/compression/dependencies/AutoGPTQ/tests/test_q4.py:1147
Methodtest_generation_with_act_order
(self)
inference/compression/dependencies/AutoGPTQ/tests/test_q4.py:1175
Functiontest_get_cumulative_document_lengths
()
tests/torch_util_test.py:6
Functiontest_get_cumulative_document_lengths
()
tests/data/util_test.py:6
Functiontest_gradient_checkpointing
(model_path: str)
tests/hf_olmo/modeling_olmo_test.py:65
Functiontest_gradient_checkpointing_disable
(model_path: str)
tests/hf_olmo/modeling_olmo_test.py:85
Methodtest_greedy_search
(self)
tests/beam_search_test.py:185
Methodtest_gumbel_sampler
(self)
tests/beam_search_test.py:457
← previousnext →3,401–3,500 of 3,821, ranked by callers