Functions — github.com/allenai/OLMo

hub / github.com/allenai/OLMo / functions

Functions3,821 in github.com/allenai/OLMo

⨍ Functions 3,821◇ Types & classes 799↳ Endpoints 9

Functionds_refactor_fn

(samples)

inference/compression/dependencies/AutoGPTQ/examples/evaluation/run_language_modeling_task.py:14

Functionds_refactor_fn

(samples)

inference/compression/dependencies/AutoGPTQ/examples/evaluation/run_sequence_classification_task.py:16

Functionds_refactor_fn

(samples)

inference/compression/dependencies/AutoGPTQ/examples/evaluation/run_text_summarization_task.py:17

Functionds_refactor_fn

(samples)

inference/compression/dependencies/AutoGPTQ/examples/peft/peft_lora_clm_instruction_tuning.py:67

Functionds_refactor_fn

(samples)

inference/compression/dependencies/AutoGPTQ/examples/peft/peft_adalora_clm_instruction_tuning.py:73

Functionds_refactor_fn

(samples)

inference/compression/dependencies/AutoGPTQ/examples/peft/peft_adaption_prompt_clm_instruction_tuning.py:67

Functiondummy_gen

()

inference/compression/dependencies/AutoGPTQ/examples/quantization/quant_with_alpaca.py:18

Functiondummy_gen

()

inference/compression/dependencies/AutoGPTQ/examples/benchmark/generation_speed.py:79

Functiondummy_init_fn

(module: torch.nn.Module)

scripts/train.py:171

Functiondump_cmd

(args: argparse.Namespace)

scripts/ladder.py:357

Methodeffective_memmap_dtype

(self)

olmo/config.py:626

Methodeffective_n_kv_heads

(self)

olmo/config.py:477

Methodeffective_n_kv_heads

(self)

hf_olmo/configuration_olmo.py:41

Methodemit

(self, record: logging.LogRecord)

olmo/util.py:240

Methodenabled

(self)

inference/compression/dependencies/AutoGPTQ/auto_gptq/quantization/quantizer.py:124

Functionensure_datasets

(beaker: Beaker, *datasets: str)

inference/efficiency/dependencies/efficiency-pentathlon/eb_gantry/util.py:213

Functionensure_entrypoint_dataset

(beaker: Beaker)

inference/efficiency/dependencies/efficiency-pentathlon/eb_gantry/util.py:120

Functionensure_github_token_secret

(beaker: Beaker, secret_name: str = constants.GITHUB_TOKEN_SECRET)

inference/efficiency/dependencies/efficiency-pentathlon/eb_gantry/util.py:177

Functionensure_repo

(allow_dirty: bool = False)

inference/efficiency/dependencies/efficiency-pentathlon/eb_gantry/util.py:105

Functionensure_workspace

( workspace: Optional[str] = None, yes: bool = False, gh_token_secret: str = GITHUB_TOKEN_SECRET,

inference/efficiency/dependencies/efficiency-pentathlon/eb_gantry/util.py:355

(tokenizer: Tokenizer)

conftest.py:82

Methodeot_token_id

(self)

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:128

Methodeot_token_id

(self)

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/models/gpt2.py:75

Methodeot_token_id

(self)

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/models/textsynth.py:58

Methodeot_token_id

(self)

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/models/gpt3.py:88

Methodeot_token_id

(self)

inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:128

Methodeot_token_id

(self)

inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/models/gpt2.py:75

Methodeot_token_id

(self)

inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/models/textsynth.py:58

Methodeot_token_id

(self)

inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/models/gpt3.py:88

Functionexact_match

Counts a question as correct if the model accurately classifies the plausibility of an answer for all candidate answers. See section 4 "Evalu

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/mc_taco.py:118

Functionexact_match

Counts a question as correct if the model accurately classifies the plausibility of an answer for all candidate answers. See section 4 "Evalu

inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/mc_taco.py:118

Functionexamine_config

Groups the token sources by their dirname and computes sizes and how much data was taken total. Prints out rows of: (token source, total_

scripts/microanneal_config_maker.py:456

Functionexamine_config

Groups the token sources by their dirname and computes sizes and how much data was taken total. Prints out rows of: (token source, total_

scripts/peteish7_config_maker.py:442

Functionexcepthook

Used to patch `sys.excepthook` in order to customize handling of uncaught exceptions.

inference/efficiency/dependencies/efficiency-pentathlon/eb_gantry/__main__.py:44

Functionexcepthook

Used to patch `sys.excepthook` in order to log exceptions.

olmo/util.py:157

Functionf1

See section 4 "Evaluation Metrics" in the paper about the F1 metric used.

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/mc_taco.py:130

Functionf1

See section 4 "Evaluation Metrics" in the paper about the F1 metric used.

inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/mc_taco.py:130

Functionf1_score

(items)

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/metrics.py:39

Methodfewshot_context

(self, doc, num_fewshot, provide_description=None, rnd=None, description=None)

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:624

Methodfewshot_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:92

Methodfewshot_context

(self, doc, num_fewshot, provide_description=None, rnd=None, description=None)

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/blimp.py:52

Methodfewshot_context

(self, doc, num_fewshot, provide_description=None, rnd=None, description=None)

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/asdiv.py:57

Methodfewshot_context

(self, doc, num_fewshot, provide_description=None, rnd=None, description=None)

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/prost.py:54

Methodfewshot_context

(self, doc, num_fewshot, provide_description=None, rnd=None, description=None)

inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:624

Methodfewshot_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:92

Methodfewshot_context

(self, doc, num_fewshot, provide_description=None, rnd=None, description=None)

inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/blimp.py:52

Methodfewshot_context

(self, doc, num_fewshot, provide_description=None, rnd=None, description=None)

inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/asdiv.py:57

Methodfewshot_context

(self, doc, num_fewshot, provide_description=None, rnd=None, description=None)

inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/prost.py:54

Methodfewshot_description

(self)

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:510

Methodfewshot_description

(self)

inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:510

Methodfewshot_examples

(self, k, rnd)

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:620

Methodfewshot_examples

(self, k, rnd)

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/cbt.py:87

Methodfewshot_examples

(self, k, rnd)

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/wsc273.py:86

Methodfewshot_examples

(self, k, rnd)

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/naturalqs.py:56

Methodfewshot_examples

(self, k, rnd)

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_test.py:151

Methodfewshot_examples

(self, k, rnd)

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_ethics.py:231

Methodfewshot_examples

(self, k, rnd)

inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:620

Methodfewshot_examples

(self, k, rnd)

inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/cbt.py:87

Methodfewshot_examples

(self, k, rnd)

inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/wsc273.py:86

Methodfewshot_examples

(self, k, rnd)

inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/naturalqs.py:56

Methodfewshot_examples

(self, k, rnd)

inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_test.py:151

Methodfewshot_examples

(self, k, rnd)

inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_ethics.py:231

Methodfewshot_instances_split

Returns the name of the split to use to find few-shot instances in.

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/task.py:123

Methodfewshot_instances_split

Returns the name of the split to use to find few-shot instances in.

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/metaicl.py:21

Methodfewshot_instances_split

Returns the name of the split to use to find few-shot instances in.

inference/efficiency/dependencies/previous_version/efficiency_benchmark/task.py:123

Methodfewshot_instances_split

Returns the name of the split to use to find few-shot instances in.

inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/metaicl.py:21

Functionfill_memmap

Write a memmap file from a file of documents.

scripts/prepare_memmap_dataset.py:226

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/mrqa.py:30

Methodfilter_subset

(example)

inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/mrqa.py:30

Methodflatten_past_key_values

( past_key_values: List[Tuple[torch.Tensor, torch.Tensor]], )

olmo/model.py:1651

Methodfn

(*args)

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:804

Methodfn

(*args)

inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:804

Methodformat_example

Question: <prompt> Choices: A. <choice1> B. <choice2> C. <choice3> D. <choice

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_test.py:129

Methodformat_example

Passage: <passage> Question: <question> Choices: A. <choice1> B. <choice2> C.

inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/logiqa.py:56

Methodformat_example

Question: <prompt> Choices: A. <choice1> B. <choice2> C. <choice3> D. <choice

inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_test.py:129

Methodformat_example

Passage: <passage> Question: <question> Choices: A. <choice1> B. <choice2> C.

inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/logiqa.py:56

Methodformat_example

(doc, keys)

olmo/eval/downstream.py:1321

Methodformat_float

(value: float)

olmo/train.py:953

Functionformat_timedelta

(td: "timedelta")

inference/efficiency/dependencies/efficiency-pentathlon/eb_gantry/util.py:190

Methodforward

(self, inp=None, **kwargs)

inference/compression/dependencies/AutoGPTQ/auto_gptq/modeling/_base.py:248

Methodforward

(self, *args, **kwargs)

inference/compression/dependencies/AutoGPTQ/auto_gptq/modeling/_base.py:450

Methodforward

(self, x: torch.Tensor)

inference/compression/dependencies/AutoGPTQ/auto_gptq/utils/peft_utils.py:63

Methodforward

(self, x: torch.Tensor)

inference/compression/dependencies/AutoGPTQ/auto_gptq/utils/peft_utils.py:217

Methodforward

(self, x)

inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/fused_llama_mlp.py:218

Methodforward

( self, hidden_states: torch.FloatTensor, layer_past: Optional[Tuple[torch.Tensor]] =

inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/fused_gptj_attn.py:147

Methodforward

Input shape: Batch x Time x Channel

inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/fused_llama_attn.py:43

Methodforward

(self, x)

inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/qlinear/qlinear_cuda_old.py:192

Methodforward

(self, x)

inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/qlinear/qlinear_triton.py:125

Methodforward

(self, x: torch.Tensor)

inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/qlinear/qlinear_cuda.py:186

Methodforward

(self, x)

inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/qlinear/qlinear_exllama.py:159

Methodforward

(ctx, input, qweight, scales, qzeros, g_idx, bits, maxq)

inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/triton_utils/kernels.py:351

Methodforward

(ctx, input, qweight, scales, qzeros, g_idx, bits, maxq)

inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/triton_utils/kernels.py:403

Methodforward

(self, inp, **kwargs)

inference/compression/dependencies/AutoGPTQ/examples/quantization/basic_usage_wikitext2.py:73

Methodforward

(self, input: torch.Tensor)

olmo/model.py:128

Methodforward

(self, x: torch.Tensor)

olmo/model.py:161

Methodforward

(self, x: torch.Tensor)

olmo/model.py:208

Methodforward

(self, x: torch.Tensor)

olmo/model.py:237

Methodforward

(self, q: torch.Tensor, k: torch.Tensor)

olmo/model.py:303

← previous next →2,301–2,400 of 3,821, ranked by callers