MCPcopy Create free account

hub / github.com/allenai/OLMo / functions

Functions3,821 in github.com/allenai/OLMo

Methodprocess_results
(self, doc, results)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_ethics.py:371
Methodprocess_results
(self, doc, results)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/arithmetic.py:69
Methodprocess_results
(self, doc, results)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_math.py:70
Methodprocess_results
(self, doc, results)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/mutual.py:86
Methodprocess_results
(self, doc, results)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/pubmedqa.py:74
Methodprocess_results
(self, doc, results)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/lambada.py:65
Methodprocess_results
(self, doc, results)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/asdiv.py:81
Methodprocess_results
Take a single document and the LM results and evaluates, returning a dict where keys are the names of submetrics and values are the values of
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/anli.py:96
Methodprocess_results
Take a single document and the LM results and evaluates, returning a dict where keys are the names of submetrics and values are the values of
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/coqa.py:143
Methodprocess_results
Take a single document and the LM results and evaluates, returning a dict where keys are the names of submetrics and values are the values of
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/mc_taco.py:84
Methodprocess_results
Take a single document and the LM results and evaluates, returning a dict where keys are the names of submetrics and values are the values of
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/quac.py:94
Methodprocess_results
Take a single document and the LM results and evaluates, returning a dict where keys are the names of submetrics and values are the values of
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/squad.py:114
Methodprocess_results
Take a single document and the LM results and evaluates, returning a dict where keys are the names of submetrics and values are the values of
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/race.py:139
Methodpush_to_hub
Upload the model to the Hugging Face Hub. Parameters: repo_id (`str`): The name of the repository you wa
inference/compression/dependencies/AutoGPTQ/auto_gptq/modeling/_base.py:462
Functionquant_fused_matmul_248_kernel
Computes: C = silu(A * B1) * (A * B2) A is of shape (M, K) float16 B is of shape (K//8, N) int32 C is of shape (M, N)
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/fused_llama_mlp.py:81
Functionquant_matmul_248_kernel
Compute the matrix multiplication C = A x B. A is of shape (M, K) float16 B is of shape (K//8, N) int32 C is of shape (M, N) float16
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/triton_utils/kernels.py:58
Functionquant_scalar
()
inference/compression/dependencies/AutoGPTQ/autogptq_extension/qigen/template.py:41
Methodquantize
( self, examples: List[Dict[str, Union[List[int], torch.LongTensor]]], batch_size: int
inference/compression/dependencies/AutoGPTQ/auto_gptq/modeling/_base.py:208
Methodquantized
(self)
inference/compression/dependencies/AutoGPTQ/auto_gptq/modeling/_base.py:153
Functionquery_openai_model
Query OpenAI chat model and save the results to output_path. `instances` is a list of dictionaries, each dictionary contains a key "prompt" a
inference/eval/utils.py:276
Functionrank0_filter
(record: logging.LogRecord)
olmo/util.py:124
Methodread
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/decontamination/archiver.py:130
Methodread
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/decontamination/archiver.py:130
Methodread_and_tell
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/decontamination/archiver.py:119
Methodread_and_tell
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/decontamination/archiver.py:119
Methodread_data
(self, plan: dist_cp.LoadPlan, planner: dist_cp.LoadPlanner)
olmo/checkpoint.py:438
Functionread_json
(path)
scripts/convert_olmo_to_hf_new.py:49
Functionread_json
(path)
scripts/convert_olmo2_to_hf.py:54
Methodread_metadata
(self)
olmo/checkpoint.py:483
Methodread_power_monitor
(ser: Serial, stop_event: threading.Event, power_readings: List[Dict])
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/efficiency/power_monitor.py:43
Methodread_power_monitor
(ser: Serial, stop_event: threading.Event, power_readings: List[Dict])
inference/efficiency/dependencies/previous_version/efficiency_benchmark/efficiency/power_monitor.py:43
Methodread_slow
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/decontamination/archiver.py:137
Methodread_slow
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/decontamination/archiver.py:137
Methodregister_contaminant
Register a string as contamination to be removed, e.g. a test set This breaks the dirt_string into ngrams to store for future cleaning
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/decontamination/janitor.py:144
Methodregister_contaminant
Register a string as contamination to be removed, e.g. a test set This breaks the dirt_string into ngrams to store for future cleaning
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/decontamination/janitor.py:144
Methodreset_lora_parameters
(self, adapter_name)
inference/compression/dependencies/AutoGPTQ/auto_gptq/utils/peft_utils.py:52
Methodreset_parameters
(self)
olmo/model.py:186
Methodreset_parameters
(self)
olmo/model.py:1052
Methodresize_token_embeddings
Resizes input token embeddings matrix of the model if `new_num_tokens != config.embedding_size`. Takes care of tying weights embeddi
hf_olmo/modeling_olmo.py:203
Methodrestore_checkpoint
Restores a checkpoint to the model and optimizer. Returns the remaining trainer state.
olmo/checkpoint.py:519
Methodrestore_checkpoint
( self, load_path: PathOrStr, dist_model: nn.Module, optim: Optimizer,
olmo/checkpoint.py:930
Methodrestore_checkpoint
( self, load_path: PathOrStr, dist_model: nn.Module, optim: Optimizer,
olmo/checkpoint.py:1014
Methodrestore_checkpoint
( self, load_path: PathOrStr, dist_model: nn.Module, optim: Optimizer,
olmo/checkpoint.py:1957
Functionrun
( cmd: Tuple[str, ...], task: str, split: str = "test", scenario: str = "accuracy", max_ba
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/__main__.py:84
Functionrun
Run an experiment on Beaker. Example: $ gantry run --name 'hello-world' -- python -c 'print("Hello, World!")'
inference/efficiency/dependencies/efficiency-pentathlon/eb_gantry/__main__.py:273
Functionrun
( cmd: Tuple[str, ...], task: str, split: str = "test", scenario: str = "accuracy", max_ba
inference/efficiency/dependencies/previous_version/efficiency_benchmark/__main__.py:84
Methodrun
(self, task_name: str, task_rename: Optional[str] = None, **kwargs)
evaluation/steps/run_catwalk.py:28
Methodrun
( self, model_path: str, model_class: Optional[str] = None, revision: Optional
evaluation/steps/run_catwalk.py:86
Methodrun
( self, outputs: Dict, **kwargs, )
evaluation/steps/run_catwalk.py:136
Methodrun
( self, model: Model, task_dict: Dict, split: Optional[str] = DEFAULT_PREDICTI
evaluation/steps/run_catwalk.py:170
Methodrun
( self, models: List[str], outputs: List[Dict], prediction_kwargs: List[Dict], gsheet: Optional[str] =
evaluation/steps/run_catwalk.py:276
Methodrun
( self, models: List[str], outputs: List[Dict], prediction_kwargs: List[Dict], gsheet: Optional[str] =
evaluation/steps/run_catwalk.py:318
Methodsameuntil_chunks
(xs, size)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/models/gpt3.py:176
Methodsameuntil_chunks
(xs, size)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/models/gpt3.py:176
Methodsample_nodes
( self, log_probs: torch.Tensor, per_node_beam_size: int, state: StateType )
olmo/beam_search.py:108
Methodsample_nodes
( self, log_probs: torch.Tensor, per_node_beam_size: int, state: StateType,
olmo/beam_search.py:316
Methodsave_checkpoint
( self, dir: PathOrStr, dist_model: nn.Module, optim: Optimizer, train
olmo/checkpoint.py:507
Methodsave_checkpoint
( self, dir: PathOrStr, dist_model: nn.Module, optim: Optimizer, train
olmo/checkpoint.py:895
Methodsave_checkpoint
( self, dir: PathOrStr, dist_model: nn.Module, optim: Optimizer, train
olmo/checkpoint.py:979
Methodsave_checkpoint
( self, dir: PathOrStr, dist_model: nn.Module, optim: Optimizer, train
olmo/checkpoint.py:1908
Methodsave_contamination_ngrams
(self, filename)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/decontamination/janitor.py:132
Methodsave_contamination_ngrams
(self, filename)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/decontamination/janitor.py:132
Functionsave_hf_dataset_to_disk
Saves a HF dataset to disk under the `datasets_dir`. It can be used to add a HF dataset to `olmo_data` as follows: ``` import datase
olmo/util.py:717
Methodsave_pretrained
alias of save_quantized
inference/compression/dependencies/AutoGPTQ/auto_gptq/modeling/_base.py:605
Methodscheduler_current
(self)
olmo/train.py:304
Methodscheduler_max
(self)
olmo/train.py:313
Methodscore
(self, predictions: torch.Tensor, log_probabilities: torch.Tensor, end_index: int)
olmo/beam_search.py:455
Methodscore
(self, predictions: torch.Tensor, log_probabilities: torch.Tensor, end_index: int)
olmo/beam_search.py:478
Methodset_activation_checkpointing
( self, strategy: Optional[ActivationCheckpointingStrategy], checkpoint_func: Optional[Callable] = Non
olmo/model.py:504
Methodset_activation_checkpointing
( self, strategy: Optional[ActivationCheckpointingStrategy], checkpoint_func: Optional[Callable] = Non
olmo/model.py:1056
Functionset_gh_token
Set or update Gantry's GitHub token for the workspace. You can create a suitable GitHub token by going to https://github.com/settings/tokens
inference/efficiency/dependencies/efficiency-pentathlon/eb_gantry/__main__.py:580
Methodset_input_embeddings
(self, value: torch.nn.Module)
hf_olmo/modeling_olmo.py:174
Methodset_output_embeddings
(self, value: torch.nn.Module)
hf_olmo/modeling_olmo.py:183
Methodset_up_storage_reader
(self, metadata: Metadata, is_coordinator: bool)
olmo/checkpoint.py:489
Methodsetup_method
(self)
tests/beam_search_test.py:108
Functionsh
(x)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/utils.py:17
Functionsh
(x)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/utils.py:17
Methodshard_size
(shard_md)
olmo/checkpoint.py:1085
Functionshort_name_for_task_object
(task: Task)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/__init__.py:774
Functionshort_name_for_task_object
(task: Task)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/__init__.py:774
Methodshould_decontaminate
Whether this task supports decontamination against model training set.
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:613
Methodshould_decontaminate
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/drop.py:111
Methodshould_decontaminate
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/sciq.py:68
Methodshould_decontaminate
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/cbt.py:77
Methodshould_decontaminate
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/triviaqa.py:57
Methodshould_decontaminate
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/piqa.py:61
Methodshould_decontaminate
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/winogrande.py:58
Methodshould_decontaminate
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/hellaswag.py:73
Methodshould_decontaminate
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/arc.py:69
Methodshould_decontaminate
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/wsc273.py:98
Methodshould_decontaminate
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/truthfulqa.py:83
Methodshould_decontaminate
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/blimp.py:72
Methodshould_decontaminate
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/unscramble.py:52
Methodshould_decontaminate
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/qa4mre.py:59
Methodshould_decontaminate
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/naturalqs.py:66
Methodshould_decontaminate
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/glue.py:77
Methodshould_decontaminate
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/superglue.py:64
Methodshould_decontaminate
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/webqs.py:58
Methodshould_decontaminate
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_test.py:163
Methodshould_decontaminate
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/translation.py:140
← previousnext →3,201–3,300 of 3,821, ranked by callers