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
↓ 7 callersMethodget_lr
(self, initial_lr: float, step: int, max_steps: int)
olmo/optim.py:718
↓ 7 callersFunctionmean
(arr)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/metrics.py:10
↓ 7 callersFunctionmean
(arr)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/metrics.py:10
↓ 7 callersFunctionprint_stderr
(*args, **kwargs)
inference/efficiency/dependencies/efficiency-pentathlon/eb_gantry/util.py:47
↓ 7 callersMethodreset
(self)
olmo/train.py:94
↓ 7 callersFunctionyesno
(x)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/metrics.py:255
↓ 7 callersFunctionyesno
(x)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/metrics.py:255
↓ 6 callersMethod__init__
( self, eleuther_task: Union[str, Callable[[], EAITask]], *, version_override:
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/eleuther.py:24
↓ 6 callersMethod__init__
( self, eleuther_task: Union[str, Callable[[], EAITask]], *, version_override:
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/eleuther.py:24
↓ 6 callersMethod_get_bucket_name_and_key
(path: str)
scripts/storage_cleaner.py:250
↓ 6 callersFunctionblock
(nu_iter, mu, tu, rho, packed, unroll, bits)
inference/compression/dependencies/AutoGPTQ/autogptq_extension/qigen/generate.py:318
↓ 6 callersFunctionbuild_sharded_checkpointer
( cfg: TrainConfig, *, name: Optional[ShardedCheckpointerType] = None, use_shared_mem_impl: bool = False )
olmo/checkpoint.py:2010
↓ 6 callersMethodclose
Closes the underlying Sqlite table. Do not use this sequence afterwards!
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tango_utils/sequences.py:368
↓ 6 callersMethodclose
Closes the underlying Sqlite table. Do not use this sequence afterwards!
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tango_utils/sequences.py:368
↓ 6 callersMethodcommit
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/decontamination/archiver.py:37
↓ 6 callersFunctionflatten_dict
Flatten a nested dictionary into a single-level dictionary. Args: dictionary (dict): The nested dictionary to be flattened.
olmo/util.py:905
↓ 6 callersFunctiongc_cuda
()
olmo/torch_util.py:142
↓ 6 callersMethodget_storage_type_for_path
(path: str)
scripts/storage_cleaner.py:113
↓ 6 callersMethodhas_validation_docs
Whether the task has a validation set
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:406
↓ 6 callersMethodhas_validation_docs
Whether the task has a validation set
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:406
↓ 6 callersMethodquantize
(self, x)
inference/compression/dependencies/AutoGPTQ/auto_gptq/quantization/quantizer.py:119
↓ 6 callersMethodstop
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/stdio_wrapper.py:147
↓ 6 callersMethodtok_encode
(self, string: str)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:152
↓ 6 callersMethodtok_encode
(self, string: str)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:152
↓ 6 callersFunctionupload
Upload source file to a target location on GCS or S3.
olmo/util.py:353
↓ 5 callersMethod_linear_warmup
(self, initial_lr: float, step: int, warmup_steps: int = 2000)
olmo/optim.py:687
↓ 5 callersMethod_save_config
(self, dir: PathOrStr, *, upload_to: Optional[str] = None)
olmo/checkpoint.py:604
↓ 5 callersMethod_temporary_wd
(self, dir: PathOrStr)
olmo/checkpoint.py:550
↓ 5 callersFunctionapply_rotary_pos_emb
(x, sincos, offset=0)
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/fused_gptj_attn.py:41
↓ 5 callersMethodextend
(self, values: Iterable[Any])
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tango_utils/sequences.py:337
↓ 5 callersFunctionget_data_path
(data_rel_path: str)
olmo_data/data.py:22
↓ 5 callersMethodhas_test_docs
Whether the task has a test set
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:411
↓ 5 callersMethodhas_test_docs
Whether the task has a test set
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:411
↓ 5 callersMethodlist_entries
Lists all the entries within the given directory. Returns only top-level entries (i.e. not entries in subdirectories). `max_file_size
scripts/storage_cleaner.py:55
↓ 5 callersMethodnew
(cls: Type[C], **kwargs)
olmo/config.py:116
↓ 5 callersMethodread
(self, file, get_meta=False, autojoin_paragraphs=True, para_joiner="\n\n")
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/decontamination/archiver.py:48
↓ 5 callersMethodrun
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/steps.py:134
↓ 5 callersMethodstop
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/stdio_wrapper.py:147
↓ 5 callersFunctionwait_for
Wait for the condition function to return True.
olmo/util.py:289
↓ 5 callersMethodwarmup
Pre-tunes the quantized kernel
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/qlinear/qlinear_triton.py:136
↓ 4 callersFunction_get_gcs_client
()
olmo/util.py:470
↓ 4 callersFunction_is_archive
(path: str, storage: StorageAdapter)
scripts/storage_cleaner.py:677
↓ 4 callersMethod_is_file
(self, bucket_name: str, key: str)
scripts/storage_cleaner.py:529
↓ 4 callersMethod_is_number
(self, text)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/drop.py:231
↓ 4 callersMethod_is_number
(self, text)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/drop.py:231
↓ 4 callersMethod_make_state_dict_compatible
Handles some cases where the state dict is valid yet may need to be transformed in order to be loaded. This modifies the sta
olmo/model.py:1781
↓ 4 callersFunction_non_meta_init_device
(config: ModelConfig)
olmo/model.py:120
↓ 4 callersMethodadd_partial
(self, attr, req, res)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:701
↓ 4 callersMethodadd_partial
(self, attr, req, res)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:701
↓ 4 callersFunctionaws_copy
(src, dest, args)
scripts/s3_unshard_to_hf.py:79
↓ 4 callersFunctionbuild_memmap_dataset
( train_config: TrainConfig, data_config: DataConfig, include_instance_metadata: bool = True )
olmo/data/__init__.py:17
↓ 4 callersFunctionbuild_optimizer
(cfg: TrainConfig, model: nn.Module)
olmo/optim.py:942
↓ 4 callersFunctionbuild_train_dataloader
( train_config: TrainConfig, *, world_size: Optional[int] = None, rank: Optional[int] = None,
olmo/data/__init__.py:86
↓ 4 callersFunctioncode_to_language
(code)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/translation.py:215
↓ 4 callersFunctioncode_to_language
(code)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/translation.py:215
↓ 4 callersFunctionconfig_from_args
(args: argparse.Namespace)
scripts/ladder.py:125
↓ 4 callersFunctiondet_hash
Returns a deterministic hash code of arbitrary Python objects. If you want to override how we calculate the deterministic hash, derive from
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tango_utils/det_hash.py:148
↓ 4 callersFunctiondet_hash
Returns a deterministic hash code of arbitrary Python objects. If you want to override how we calculate the deterministic hash, derive from
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tango_utils/det_hash.py:148
↓ 4 callersMethoddevice
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:148
↓ 4 callersFunctionefficiency_benchmark_mt_conversion
(**kwargs)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/efficiency_benchmark.py:174
↓ 4 callersFunctionefficiency_benchmark_mt_conversion
(**kwargs)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/efficiency_benchmark.py:174
↓ 4 callersMethodencode_batch
Encode a batch of strings into token IDs.
olmo/tokenizer.py:175
↓ 4 callersFunctionfind_latest_checkpoint
(dir: PathOrStr)
olmo/util.py:393
↓ 4 callersFunctionfind_layers
(module, layers=None, name="")
inference/compression/dependencies/AutoGPTQ/auto_gptq/modeling/_utils.py:28
↓ 4 callersFunctionformat_value_and_unit
(value: int, unit: str)
inference/efficiency/dependencies/efficiency-pentathlon/eb_gantry/util.py:191
↓ 4 callersFunctiongen_and_compile
(n, m, t, nb, mb, tb, nu, mu, tu, p, unroll, bits=4, gs=-1, module=False)
inference/compression/dependencies/AutoGPTQ/autogptq_extension/qigen/generate.py:786
↓ 4 callersFunctiongeneral_detokenize
(string)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/utils.py:65
↓ 4 callersFunctiongeneral_detokenize
(string)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/utils.py:65
↓ 4 callersFunctionget_causal_attention_bias
(cache: BufferCache, seq_len: int, device: torch.device)
olmo/model.py:380
↓ 4 callersFunctionget_document_lengths
(input_ids: torch.Tensor, eos_token_id: int)
olmo/data/util.py:122
↓ 4 callersMethodget_max_grad_norm
( self, initial_max_grad_norm: Optional[float], step: int, max_steps: int )
olmo/optim.py:677
↓ 4 callersFunctionget_module_by_name_suffix
(model, module_name: str)
inference/compression/dependencies/AutoGPTQ/auto_gptq/modeling/_utils.py:46
↓ 4 callersMethodget_original
(self, newarr)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/utils.py:135
↓ 4 callersMethodget_original
(self, newarr)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/utils.py:135
↓ 4 callersMethodget_reordered
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/utils.py:132
↓ 4 callersMethodget_reordered
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/utils.py:132
↓ 4 callersFunctionget_state_with_grads
(model)
tests/grad_norm_test.py:132
↓ 4 callersFunctionis_non_str_iterable
(obj)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/metrics.py:157
↓ 4 callersFunctionis_non_str_iterable
(obj)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/metrics.py:157
↓ 4 callersFunctionis_url
(path: PathOrStr)
olmo/util.py:298
↓ 4 callersMethodlast_problem
(cls, doc)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/race.py:100
↓ 4 callersMethodlast_problem
(cls, doc)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/race.py:100
↓ 4 callersFunctionopt_at
(opt, idx, key)
tests/checkpoint_test.py:16
↓ 4 callersFunctionopt_at
(opt, idx, key)
tests/ddp_checkpoint_test.py:16
↓ 4 callersMethodpad_tokens_until_max
truncate from left if len(tokens) > model_ctx_len, max_len is not considered then queries are already truncated at max length of model_ctx_len
olmo/eval/downstream.py:278
↓ 4 callersMethodpreprocess
(cls, text)
olmo/eval/downstream.py:476
↓ 4 callersMethodreset_parameters
(self)
olmo/model.py:479
↓ 4 callersMethodstart
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/stdio_wrapper.py:140
↓ 4 callersMethodupdate_with
(self, **kwargs)
olmo/config.py:161
↓ 4 callersMethodwarmup_triton
(self, enabled: bool = True)
inference/compression/dependencies/AutoGPTQ/auto_gptq/modeling/_base.py:985
↓ 3 callersMethod__init__
(self, resource: str)
scripts/storage_cleaner.py:1293
↓ 3 callersMethod__init__
(self, cfg: TrainConfig, thread_count: Optional[int] = None)
olmo/checkpoint.py:502
↓ 3 callersMethod_cast_if_autocast_enabled
(self, tensor: torch.Tensor, dtype: Optional[torch.dtype] = None)
olmo/model.py:175
↓ 3 callersMethod_clean_param_name
(self, name: str)
olmo/optim.py:45
↓ 3 callersMethod_collate_data
(self, set)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/race.py:54
↓ 3 callersMethod_collate_data
(self, set)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/race.py:54
↓ 3 callersFunction_format_dir_or_archive_path
(storage: StorageAdapter, path: str)
scripts/storage_cleaner.py:682
↓ 3 callersFunction_get_data_traversable
(data_rel_path: str)
olmo_data/data.py:9
↓ 3 callersMethod_get_max_grad_norm_coeff
( self, initial_value: Optional[float], step: int, max_steps: int )
olmo/optim.py:662
↓ 3 callersFunction_is_checkpoint_dir
(directory: str)
scripts/storage_cleaner.py:768
← previousnext →101–200 of 3,821, ranked by callers