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
Method__init__
(self, bits, group_size, infeatures, outfeatures, bias, trainable=False, **kwargs)
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/qlinear/qlinear_exllama.py:44
Method__init__
( self, fn, arg_names, configs, key, reset_to_zero, pr
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/triton_utils/custom_autotune.py:15
Method__init__
( self, model, tokenizer: PreTrainedTokenizer, classes: List[str], dat
inference/compression/dependencies/AutoGPTQ/auto_gptq/eval_tasks/sequence_classification_task.py:33
Method__init__
( self, model: Union[BaseGPTQForCausalLM, PreTrainedModel], tokenizer: PreTrainedToken
inference/compression/dependencies/AutoGPTQ/auto_gptq/eval_tasks/_base.py:12
Method__init__
( self, model, tokenizer, data_name_or_path: str, prompt_col_name: str
inference/compression/dependencies/AutoGPTQ/auto_gptq/eval_tasks/language_modeling_task.py:10
Method__init__
( self, model, tokenizer, data_name_or_path: str, prompt_col_name: str
inference/compression/dependencies/AutoGPTQ/auto_gptq/eval_tasks/text_summarization_task.py:12
Method__init__
(self, module)
inference/compression/dependencies/AutoGPTQ/examples/quantization/basic_usage_wikitext2.py:69
Method__init__
( self, min_new_tokens: int = None, eos_token_id: int = None, )
inference/compression/dependencies/AutoGPTQ/examples/benchmark/generation_speed.py:22
Method__init__
(self)
scripts/storage_cleaner.py:129
Method__init__
(self)
scripts/storage_cleaner.py:229
Method__init__
(self, storage_type: StorageType)
scripts/storage_cleaner.py:393
Method__init__
Create a new memmap file. Args: path (str): Location for the memmap file. If the path is not local, the memmap file will be
scripts/prepare_memmap_dataset.py:115
Method__init__
(self, *args, record_update_metrics: bool = False, selective_updates: bool = False, **kwargs)
olmo/optim.py:39
Method__init__
(self, *args, record_update_metrics: bool = False, selective_updates: bool = False, **kwargs)
olmo/optim.py:507
Method__init__
( self, base_tokenizer: BaseTokenizer, eos_token_id: int, pad_token_id: Option
olmo/tokenizer.py:30
Method__init__
( self, temperature: float = 1.0, with_replacement: bool = False, )
olmo/beam_search.py:127
Method__init__
( self, k: int = 1, temperature: float = 1.0, with_replacement: bool = False,
olmo/beam_search.py:161
Method__init__
( self, p: float = 0.9, temperature: float = 1.0, with_replacement: bool = Fal
olmo/beam_search.py:226
Method__init__
(self, temperature: float = 1.0)
olmo/beam_search.py:302
Method__init__
(self, length_penalty: float = 1.0)
olmo/beam_search.py:474
Method__init__
(self, ngram_size: int, **kwargs)
olmo/beam_search.py:594
Method__init__
( self, path: PathOrStr, single_file_per_rank: bool = True, sync_files: bool =
olmo/checkpoint.py:339
Method__init__
( self, path: PathOrStr, *, local_cache: Optional[PathOrStr] = None, thread_count: Optional[int] = Non
olmo/checkpoint.py:415
Method__init__
(self, cfg: TrainConfig, thread_count: Optional[int] = None, use_shared_mem_impl: bool = False)
olmo/checkpoint.py:975
Method__init__
( self, *, level: Union[int, str] = logging.NOTSET, console: Optional[Console]
olmo/util.py:228
Method__init__
( self, config: ModelConfig, *, size: Optional[int] = None, elementwis
olmo/model.py:136
Method__init__
( self, config: ModelConfig, size: Optional[int] = None, low_precision: bool =
olmo/model.py:198
Method__init__
( self, config: ModelConfig, size: Optional[int] = None, elementwise_affine: O
olmo/model.py:229
Method__init__
(self, config: ModelConfig, cache: BufferCache)
olmo/model.py:259
Method__init__
(self, config: ModelConfig)
olmo/model.py:324
Method__init__
(self, layer_id: int, config: ModelConfig, cache: BufferCache)
olmo/model.py:412
Method__init__
(self, layer_id: int, config: ModelConfig, cache: BufferCache)
olmo/model.py:679
Method__init__
(self, layer_id: int, config: ModelConfig, cache: BufferCache)
olmo/model.py:828
Method__init__
(self, config: ModelConfig, layer_offset: int, modules: Optional[Iterable[nn.Module]] = None)
olmo/model.py:1006
Method__init__
( self, *paths: PathOrStr, chunk_size: int = 1024, memmap_dtype: Union[Type[np
olmo/data/memmap_dataset.py:46
Method__init__
( self, dataset: Union[Sequence[List[int]], Sequence[torch.Tensor], Sequence[Dict[str, Any]]],
olmo/data/iterable_dataset.py:27
Method__init__
metric_type: f1, acc, len_norm, pmi_dc, ce_loss, bpb
olmo/eval/downstream.py:27
Method__init__
( self, tokenizer: Tokenizer, dataset_path: str, dataset_name: Union[str, Sequ
olmo/eval/downstream.py:168
Method__init__
( self, tokenizer, dataset_path="piqa", dataset_name="plain_text", )
olmo/eval/downstream.py:420
Method__init__
( self, tokenizer, dataset_path="hellaswag", dataset_name=None, )
olmo/eval/downstream.py:463
Method__init__
( self, tokenizer, dataset_path="winogrande", dataset_name="winogrande_xl",
olmo/eval/downstream.py:526
Method__init__
( self, tokenizer, dataset_path="openbookqa", dataset_name="main", )
olmo/eval/downstream.py:628
Method__init__
( self, tokenizer, dataset_path="boolq", dataset_name=None, )
olmo/eval/downstream.py:668
Method__init__
( self, tokenizer, dataset_path="sciq", dataset_name=None, )
olmo/eval/downstream.py:718
Method__init__
( self, tokenizer, dataset_path="ai2_arc", dataset_name="ARC-Easy", )
olmo/eval/downstream.py:765
Method__init__
( self, tokenizer, dataset_path="ai2_arc", dataset_name="ARC-Challenge", )
olmo/eval/downstream.py:805
Method__init__
( self, tokenizer, dataset_path="allenai/basic_arithmetic", dataset_name=None,
olmo/eval/downstream.py:845
Method__init__
( self, tokenizer, dataset_path="tau/commonsense_qa", dataset_name=None, )
olmo/eval/downstream.py:871
Method__init__
( self, tokenizer, dataset_path="social_i_qa", dataset_name=None, )
olmo/eval/downstream.py:896
Method__init__
( self, tokenizer, dataset_path="super_glue", dataset_name="copa", )
olmo/eval/downstream.py:948
Method__init__
( self, tokenizer, dataset_path="super_glue", dataset_name="cb", )
olmo/eval/downstream.py:1039
Method__init__
( self, tokenizer, dataset_path="glue", dataset_name="mrpc", )
olmo/eval/downstream.py:1081
Method__init__
( self, tokenizer, dataset_path="glue", dataset_name="sst2", )
olmo/eval/downstream.py:1151
Method__init__
( self, tokenizer, dataset_path="hails/mmlu_no_train", dataset_name=None,
olmo/eval/downstream.py:1274
Method__init__
( self, tokenizer, dataset_path="trivia_qa", dataset_name="rc.wikipedia.nocont
olmo/eval/downstream.py:1396
Method__init__
( self, tokenizer, dataset_path="nq_open", dataset_name=None, )
olmo/eval/downstream.py:1433
Method__init__
( self, tokenizer: Tokenizer, dataset_path: str, dataset_name: Union[str, Sequ
olmo/eval/downstream.py:1462
Method__init__
(self, use_cache: bool = False, **kwargs)
hf_olmo/configuration_olmo.py:18
Method__init__
(self, config: OLMoConfig, model: Optional[OLMo] = None, init_params: bool = False)
hf_olmo/modeling_olmo.py:53
Method__iter__
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/metrics/accuracy.py:13
Method__iter__
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:784
Method__iter__
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/metrics/accuracy.py:13
Method__iter__
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:784
Method__iter__
(self)
olmo/data/iterable_dataset.py:126
Method__len__
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/utils.py:40
Method__len__
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tango_utils/sequences.py:61
Method__len__
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tango_utils/sequences.py:155
Method__len__
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tango_utils/sequences.py:233
Method__len__
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tango_utils/sequences.py:355
Method__len__
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/utils.py:40
Method__len__
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tango_utils/sequences.py:61
Method__len__
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tango_utils/sequences.py:155
Method__len__
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tango_utils/sequences.py:233
Method__len__
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tango_utils/sequences.py:355
Method__len__
Length of the memmap file in tokens that have been written.
scripts/prepare_memmap_dataset.py:137
Method__len__
(self)
olmo/data/memmap_dataset.py:174
Method__len__
(self)
olmo/eval/downstream.py:209
Method__new__
(cls, *args, **kwargs)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tango_utils/det_hash.py:50
Method__new__
(cls, *args, **kwargs)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tango_utils/det_hash.py:50
Method__post_init__
(self)
inference/compression/dependencies/AutoGPTQ/auto_gptq/modeling/_base.py:55
Method__post_init__
(self)
olmo/config.py:536
Method__post_init__
(self)
olmo/train.py:236
Method__repr__
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:798
Method__repr__
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:798
Method__repr__
(self)
olmo/util.py:61
Method__rrshift__
(self, other)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/race.py:33
Method__rrshift__
(self, other)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/race.py:33
Method__setitem__
(self, i: Union[int, slice], value: Any)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tango_utils/sequences.py:307
Method__setitem__
(self, i: Union[int, slice], value: Any)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tango_utils/sequences.py:307
Method__str__
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/translation.py:203
Method__str__
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/translation.py:203
Method__str__
(self)
olmo/util.py:58
Method_collate
(x)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:228
Method_collate
(x)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/models/gpt3.py:119
Method_collate
(x)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:228
Method_collate
(x)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/models/gpt3.py:119
Method_convert_tensor_to_list
(tensor)
inference/compression/dependencies/AutoGPTQ/auto_gptq/modeling/_base.py:173
Method_copy_sharded_data_to_shared_mem
(self, world_size: int, shard_filepath: Path)
olmo/checkpoint.py:1135
Method_field_ordering
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/raft.py:50
Method_field_ordering
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/raft.py:50
← previousnext →1,301–1,400 of 3,821, ranked by callers