MCPcopy Create free account

hub / github.com/allenai/OLMo / functions

Functions3,821 in github.com/allenai/OLMo

↓ 1 callersMethodinit_state
( self, batch_size: int, )
olmo/beam_search.py:598
↓ 1 callersFunctioninspect_data_without_device_data_indices
( run_path: str, *steps: int, world_size: int, ranks: List[int], reference_step: int,
scripts/inspect_train_data.py:60
↓ 1 callersFunctioninstall_excepthook
()
olmo/util.py:171
↓ 1 callersMethodinstance_as_rank_classification
( self, instance: Dict[str, Any], *, include_instructions: bool = False,
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/raft.py:70
↓ 1 callersMethodinstance_as_rank_classification
( self, instance: Dict[str, Any], *, include_instructions: bool = False,
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/raft.py:70
↓ 1 callersMethodinstance_to_eleuther_context
(self, instance: Dict[str, Any], *, num_fewshot: int = 0)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/eleuther.py:97
↓ 1 callersMethodinstance_to_eleuther_context
(self, instance: Dict[str, Any], *, num_fewshot: int = 0)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/eleuther.py:97
↓ 1 callersMethodis_equiv
(self, str1, str2, verbose=False)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_math.py:88
↓ 1 callersMethodis_equiv
(self, str1, str2, verbose=False)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_math.py:88
↓ 1 callersFunctionlevenshtein_distance
(seq1: Sequence, seq2: Sequence)
inference/compression/dependencies/AutoGPTQ/auto_gptq/eval_tasks/_utils/classification_utils.py:7
↓ 1 callersFunctionlist_oe_eval_tasks
(num_examples=0)
scripts/list_evals_from_oe_eval.py:46
↓ 1 callersMethodload_checkpoint
( self, load_path: PathOrStr, *, local_cache: Optional[PathOrStr] = None,
olmo/checkpoint.py:874
↓ 1 callersFunctionload_data
(data_path, tokenizer, n_samples)
inference/compression/dependencies/AutoGPTQ/examples/quantization/quant_with_alpaca.py:12
↓ 1 callersFunctionload_data
(data_path, tokenizer, n_samples, max_new_tokens)
inference/compression/dependencies/AutoGPTQ/examples/benchmark/generation_speed.py:73
↓ 1 callersFunctionload_fsdp_model_and_optim_state
Use this to load a state dict for an FSDP model and its optimizer via :module:`torch.distributed.checkpoint` functions. This should be used d
olmo/checkpoint.py:149
↓ 1 callersFunctionload_hf_lm_and_tokenizer
( model_name_or_path, tokenizer_name_or_path=None, device_map="auto", load_in_8bit=False,
inference/eval/utils.py:158
↓ 1 callersFunctionload_model_state
Load model state from a distributed FSDP model checkpoint created from :func:`save_fsdp_model_and_optim_state()`. Note that ``model`` should
olmo/checkpoint.py:319
↓ 1 callersFunctionload_model_tokenizer
( model_name_or_path: str, tokenizer_name_or_path: Optional[str] = None, from_pretrained: bool = F
inference/compression/dependencies/AutoGPTQ/examples/benchmark/generation_speed.py:131
↓ 1 callersMethodloglikelihood_rolling
Compute full log-likelihood of a string, with no truncation, for perplexity computation - We will use the full max context length of the model
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:54
↓ 1 callersMethodloglikelihood_rolling
Compute full log-likelihood of a string, with no truncation, for perplexity computation - We will use the full max context length of the model
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:54
↓ 1 callersFunctionlongest_common_prefix
Finds the longest common prefix among a list of strings.
hf_olmo/convert_olmo_to_hf.py:33
↓ 1 callersFunctionlower
(text)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/qasper.py:59
↓ 1 callersFunctionlower
(text)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/qasper.py:59
↓ 1 callersFunctionmain
()
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/__main__.py:31
↓ 1 callersFunctionmain
()
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/efficiency/power_monitor.py:78
↓ 1 callersFunctionmain
()
inference/efficiency/dependencies/efficiency-pentathlon/eb_gantry/__main__.py:68
↓ 1 callersFunctionmain
()
inference/efficiency/dependencies/previous_version/efficiency_benchmark/__main__.py:31
↓ 1 callersFunctionmain
()
inference/efficiency/dependencies/previous_version/efficiency_benchmark/efficiency/power_monitor.py:78
↓ 1 callersFunctionmain
(args)
inference/eval/mmlu/run_eval.py:144
↓ 1 callersFunctionmain
Run quantization.
inference/compression/run_quantization.py:62
↓ 1 callersFunctionmain
()
inference/compression/dependencies/AutoGPTQ/examples/quantization/quant_with_alpaca.py:70
↓ 1 callersFunctionmain
()
inference/compression/dependencies/AutoGPTQ/examples/quantization/basic_usage.py:10
↓ 1 callersFunctionmain
()
inference/compression/dependencies/AutoGPTQ/examples/quantization/basic_usage_gpt_xl.py:37
↓ 1 callersFunctionmain
()
inference/compression/dependencies/AutoGPTQ/examples/quantization/basic_usage_wikitext2.py:138
↓ 1 callersFunctionmain
()
inference/compression/dependencies/AutoGPTQ/examples/benchmark/generation_speed.py:220
↓ 1 callersFunctionmain
()
inference/compression/dependencies/AutoGPTQ/examples/evaluation/run_language_modeling_task.py:31
↓ 1 callersFunctionmain
()
inference/compression/dependencies/AutoGPTQ/examples/evaluation/run_sequence_classification_task.py:29
↓ 1 callersFunctionmain
()
inference/compression/dependencies/AutoGPTQ/examples/evaluation/run_text_summarization_task.py:28
↓ 1 callersFunctionmain
()
scripts/compare_module_outputs.py:131
↓ 1 callersFunctionmain
()
scripts/compare_model_state.py:89
↓ 1 callersFunctionmain
()
scripts/storage_cleaner.py:1497
↓ 1 callersFunctionmain
(input_dir: str, input_ext: str, output_dir: str, output_ext: str, size_limit: int)
scripts/split_gz.py:23
↓ 1 callersFunctionmain
( input_dir: Union[str, Path], output_dir: Union[str, Path], sharded_checkpoint_type: Optional[Sha
scripts/unshard.py:15
↓ 1 callersFunctionmain
()
scripts/convert_olmo_to_hf_new.py:232
↓ 1 callersFunctionmain
()
scripts/s3_unshard_to_hf.py:160
↓ 1 callersFunctionmain
()
scripts/convert_olmo2_to_hf.py:253
↓ 1 callersFunctionmain
( wandb_run_path: str, new_group_name: str, )
scripts/wandb_change_group.py:19
↓ 1 callersFunctionmain
(wandb_run_path: str, step: int)
scripts/learning_rate_at_step_from_wandb.py:19
↓ 1 callersFunctionmain
(opts)
scripts/prepare_tulu_data.py:20
↓ 1 callersFunctionmain
( src: Tuple[str, ...], output: str, tokenizer_id: str = "EleutherAI/gpt-neox-20b", dtype_str:
scripts/prepare_memmap_dataset.py:378
↓ 1 callersFunctionmain
(cfg: TrainConfig, output_dir: Path)
scripts/run_dataloader.py:21
↓ 1 callersFunctionmain
(save_path: Path, args_list: List[str])
scripts/init_config.py:16
↓ 1 callersFunctionmain
(input: PathOrStr, output: PathOrStr)
scripts/convert_pt_to_safetensors.py:9
↓ 1 callersFunctionmain
()
scripts/list_evals_from_oe_eval.py:71
↓ 1 callersFunctionmain
()
scripts/release_notes.py:75
↓ 1 callersFunctionmain
()
scripts/prepare_changelog.py:7
↓ 1 callersFunctionmain
( left_run_path: str, right_run_path: str, )
scripts/compare_wandb_configs.py:91
↓ 1 callersFunctionmain
(wandb_run_path: str)
scripts/group_name_from_wandb.py:15
↓ 1 callersFunctionmain
(cfg: TrainConfig)
scripts/show_model_size.py:20
↓ 1 callersFunctionmain
( run_path: str, *steps: int, world_size: Optional[int] = None, rank: Optional[int] = None,
scripts/inspect_train_data.py:121
↓ 1 callersFunctionmain
()
hf_olmo/convert_olmo_to_hf.py:248
↓ 1 callersFunctionmake_local_checkpoint
(checkpoint_dir: str)
hf_olmo/convert_olmo_to_hf.py:178
↓ 1 callersMethodmake_metrics
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/task.py:130
↓ 1 callersMethodmake_metrics
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/task.py:130
↓ 1 callersFunctionmake_parser
()
scripts/s3_unshard_to_hf.py:24
↓ 1 callersFunctionmake_source_and_target
Recursively list all files in the source directories and create a corresponding list of destination.
scripts/prepare_memmap_dataset.py:300
↓ 1 callersMethodmake_sure_compatible_with_peft
( model: PreTrainedModel, use_triton: bool, desc_act: bool, group_size: int, bits: int )
inference/compression/dependencies/AutoGPTQ/auto_gptq/modeling/_base.py:1010
↓ 1 callersFunctionmake_sure_no_tensor_in_meta_device
(model, use_triton, desc_act, group_size, bits: int)
inference/compression/dependencies/AutoGPTQ/auto_gptq/modeling/_utils.py:287
↓ 1 callersFunctionmaybe_unshard
(checkpoint_dir: str)
hf_olmo/convert_olmo_to_hf.py:235
↓ 1 callersFunctionmean_stderr
(arr)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/metrics.py:24
↓ 1 callersFunctionmean_stderr
(arr)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/metrics.py:24
↓ 1 callersFunctionmem_model
(N, M, T, mu, tu, bits, l1, p, gs, verbose=False)
inference/compression/dependencies/AutoGPTQ/autogptq_extension/qigen/generate.py:13
↓ 1 callersFunctionmove_run
(run_path: str, dest_dir: str, config: MoveRunConfig)
scripts/storage_cleaner.py:1275
↓ 1 callersFunctionnormalize_answers
(answer: Any, answer_mappings: Optional[Dict[str, int]] = None)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/huggingface.py:114
↓ 1 callersFunctionnormalize_answers
(answer: Any, answer_mappings: Optional[Dict[str, int]] = None)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/huggingface.py:114
↓ 1 callersMethodnum_special_tokens_to_add
(self, is_pair: bool = False)
olmo/tokenizer.py:156
↓ 1 callersMethodoffline_output_path
(self, base_dir: str)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/efficiency_benchmark.py:53
↓ 1 callersMethodoffline_output_path
(self, base_dir: str)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/efficiency_benchmark.py:53
↓ 1 callersFunctionopt_eval
(model, testenc, dev, seqlen=2048)
inference/compression/dependencies/AutoGPTQ/examples/quantization/basic_usage_wikitext2.py:46
↓ 1 callersMethodpack
(self, linear, scales, zeros, g_idx=None)
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/qlinear/qlinear_cuda.py:89
↓ 1 callersFunctionpack_model
( model, quantizers, bits, group_size, use_triton=False, use_cuda_fp16=True, desc_
inference/compression/dependencies/AutoGPTQ/auto_gptq/modeling/_utils.py:114
↓ 1 callersMethodparse_answer
(cls, answer)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/drop.py:100
↓ 1 callersMethodparse_answer
(cls, answer)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/drop.py:100
↓ 1 callersFunctionparse_git_remote_url
Parse a git remote URL into a GitHub (account, repo) pair. :raises InvalidRemoteError: If the URL can't be parsed correctly.
inference/efficiency/dependencies/efficiency-pentathlon/eb_gantry/util.py:55
↓ 1 callersFunctionparse_length
(length: str, model_size: int)
scripts/ladder.py:107
↓ 1 callersFunctionparse_requirements_file
(path)
inference/efficiency/dependencies/efficiency-pentathlon/setup.py:4
↓ 1 callersFunctionparse_size
(size: str)
scripts/ladder.py:93
↓ 1 callersFunctionperform_operation
(args: argparse.Namespace)
scripts/storage_cleaner.py:1316
↓ 1 callersMethodpost_init
(self)
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/qlinear/qlinear_cuda.py:86
↓ 1 callersMethodpredict
(self, inputs)
inference/efficiency/dependencies/previous_version/olmo_efficiency.py:54
↓ 1 callersMethodprep_examples
Append doc_ids to each example so that they are processed together in the metric
olmo/eval/downstream.py:212
↓ 1 callersMethodprep_examples
(self)
olmo/eval/downstream.py:1509
↓ 1 callersMethodprepare_offline_instances
(self, base_dir: str, split: str, override: bool = True)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/efficiency_benchmark.py:93
↓ 1 callersMethodprepare_offline_instances
(self, base_dir: str, split: str, override: bool = True)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/efficiency_benchmark.py:93
↓ 1 callersMethodpreprocess
(cls, string: str)
olmo/eval/downstream.py:1164
↓ 1 callersFunctionprint_data_differences
(left_data_paths: Counter, right_data_paths: Counter)
scripts/compare_wandb_configs.py:69
↓ 1 callersFunctionprint_parameters_module
(bits, mu, nu, tu, unroll, p, gs=-1)
inference/compression/dependencies/AutoGPTQ/autogptq_extension/qigen/generate.py:102
↓ 1 callersMethodprocess
(self, output_batches: Iterable[str])
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/steps.py:183
↓ 1 callersMethodprocess
(self, output_batches: Iterable[str])
inference/efficiency/dependencies/previous_version/efficiency_benchmark/steps.py:177
↓ 1 callersMethodprocess_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/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:480
← previousnext →901–1,000 of 3,821, ranked by callers