MCPcopy Create free account

hub / github.com/allenai/OLMo / functions

Functions3,821 in github.com/allenai/OLMo

Methodforward
(self, x: torch.Tensor)
olmo/model.py:329
Methodforward
(self, x: torch.Tensor)
olmo/model.py:362
Methodforward
( self, x: torch.Tensor, attention_bias: Optional[torch.FloatTensor] = None, l
olmo/model.py:651
Methodforward
( self, x: torch.Tensor, attention_bias: Optional[torch.Tensor] = None, layer_
olmo/model.py:722
Methodforward
( self, x: torch.Tensor, attention_bias: Optional[torch.Tensor] = None, layer_
olmo/model.py:915
Methodforward
( self, x: torch.Tensor, attention_bias: Optional[torch.FloatTensor] = None, l
olmo/model.py:1013
Methodforward
:param input_ids: A tensor of shape `(batch_size, seq_len)`. :param input_embeddings: A tensor of shape `(batch_size, seq_len, d_mode
olmo/model.py:1247
Methodforward
( self, input_ids: torch.LongTensor = None, inputs_embeds: Optional[torch.FloatTensor]
hf_olmo/modeling_olmo.py:84
Functionforward_module
(nu, mu, tu, p, unroll, bits)
inference/compression/dependencies/AutoGPTQ/autogptq_extension/qigen/generate.py:1139
Functionforward_module_gs
(nu, mu, tu, p, unroll, bits)
inference/compression/dependencies/AutoGPTQ/autogptq_extension/qigen/generate.py:967
Methodfrom_checkpoint
Load a tokenizer from a checkpoint.
olmo/tokenizer.py:112
Methodfrom_pretrained
load un-quantized pretrained model to cpu
inference/compression/dependencies/AutoGPTQ/auto_gptq/modeling/_base.py:617
Methodfrom_pretrained
( cls, pretrained_model_name_or_path: str, quantize_config: BaseQuantizeConfig,
inference/compression/dependencies/AutoGPTQ/auto_gptq/modeling/auto.py:49
Methodfrom_quantized
load quantized model from local disk
inference/compression/dependencies/AutoGPTQ/auto_gptq/modeling/_base.py:716
Methodfsdp_precision
(self)
olmo/config.py:1266
Methodfsdp_wrap_fn
(module, recurse: bool = True, nonwrapped_numel: int = 0)
olmo/model.py:1477
Methodgcs_client
(self)
scripts/storage_cleaner.py:243
Functiongenerate_completions
( model, tokenizer, prompts, batch_size=1, stop_id_sequences=None, disable_tqdm=False, **generation_kwargs
inference/eval/utils.py:34
Methodget_answer_choice
(self, raw_text)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/coqa.py:88
Methodget_answer_choice
(self, raw_text)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/coqa.py:88
Methodget_bytes_range
(self, index: int, length: int)
olmo/util.py:898
Methodget_etag
(self)
olmo/util.py:885
Methodget_fewshot_instances
( self, num_shots: int, *, exceptions: Union[None, Dict[str, Any], Iterable[Di
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/task.py:139
Methodget_fewshot_instances
( self, num_shots: int, *, exceptions: Union[None, Dict[str, Any], Iterable[Di
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/metaicl.py:42
Methodget_fewshot_instances
( self, num_shots: int, *, exceptions: Union[None, Dict[str, Any], Iterable[Di
inference/efficiency/dependencies/previous_version/efficiency_benchmark/task.py:139
Methodget_fewshot_instances
( self, num_shots: int, *, exceptions: Union[None, Dict[str, Any], Iterable[Di
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/metaicl.py:42
Methodget_file_size
(self, path: str)
scripts/storage_cleaner.py:198
Methodget_file_size
(self, path: str)
scripts/storage_cleaner.py:354
Methodget_file_size
(self, path: str)
scripts/storage_cleaner.py:547
Methodget_input_embeddings
(self)
hf_olmo/modeling_olmo.py:171
Methodget_lls
(targets)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/truthfulqa.py:108
Methodget_lls
(targets)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/truthfulqa.py:108
Methodget_lr
(self, initial_lr: float, step: int, max_steps: int)
olmo/optim.py:699
Methodget_lr
(self, initial_lr: float, step: int, max_steps: int)
olmo/optim.py:735
Methodget_lr
(self, initial_lr: float, step: int, max_steps: int)
olmo/optim.py:747
Methodget_lr
(self, initial_lr: float, step: int, max_steps: int)
olmo/optim.py:770
Methodget_lr
(self, initial_lr: float, step: int, max_steps: int)
olmo/optim.py:787
Methodget_lr
(self, initial_lr: float, step: int, max_steps: int)
olmo/optim.py:799
Methodget_lr
(self, initial_lr: float, step: int, max_steps: int)
olmo/optim.py:819
Methodget_max_grad_norm_ratio
( self, initial_max_grad_norm_ratio: Optional[float], step: int, max_steps: int )
olmo/optim.py:682
Functionget_model
(model_name)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/models/__init__.py:12
Functionget_model
(model_name)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/models/__init__.py:12
Functionget_model_path
( model_path: Union[str, os.PathLike], revision: Optional[str] = None, )
evaluation/steps/get_model.py:15
Methodget_output_embeddings
(self)
hf_olmo/modeling_olmo.py:177
Methodget_post_step_metrics
( self, module: nn.Module, process_group: Optional[dist.ProcessGroup] = None )
olmo/optim.py:354
Methodget_post_step_metrics
( self, module: nn.Module, process_group: Optional[dist.ProcessGroup] = None )
olmo/optim.py:610
Methodget_random_batch_instances
(self, split: str)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/efficiency_benchmark.py:90
Methodget_random_batch_instances
(self, split: str)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/efficiency_benchmark.py:90
Methodget_resource
(self, temp_file: io.BufferedWriter)
olmo/util.py:895
Functionget_rolling_token_windows
- context_len allows for a rolling window context, allowing each prediction window to potentially condition on some context :param tok
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/utils.py:75
Functionget_rolling_token_windows
- context_len allows for a rolling window context, allowing each prediction window to potentially condition on some context :param tok
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/utils.py:75
Methodget_single_stream_instances
(self, split: str)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/efficiency_benchmark.py:87
Methodget_single_stream_instances
(self, split: str)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/efficiency_benchmark.py:87
Methodget_size
(self)
olmo/util.py:890
Methodget_split
(self, split: str)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/metaicl.py:34
Methodget_split
(self, split: str)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/eleuther.py:77
Methodget_split
(self, split: str)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/eleuther.py:209
Methodget_split
(self, split: str)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/eleuther.py:242
Methodget_split
(self, split: str)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/efficiency_benchmark.py:155
Methodget_split
(self, split: str)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/huggingface.py:72
Methodget_split
(self, split: str)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/metaicl.py:34
Methodget_split
(self, split: str)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/eleuther.py:77
Methodget_split
(self, split: str)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/eleuther.py:209
Methodget_split
(self, split: str)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/eleuther.py:242
Methodget_split
(self, split: str)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/efficiency_benchmark.py:155
Methodget_split
(self, split: str)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/huggingface.py:72
Methodget_state_for_param
(self, param: nn.Parameter)
olmo/optim.py:607
Functionget_task_dict
(task_name_list: List[Union[str, efficiency_benchmark.dependencies.lm_eval.base.Task]])
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/__init__.py:315
Functionget_task_dict
(task_name_list: List[Union[str, efficiency_benchmark.dependencies.lm_eval.base.Task]])
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/__init__.py:315
Functionget_train_overlap_stub
(docs, ngrams_path, ngrams_n_size)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/decontamination/decontaminate.py:14
Functionget_train_overlap_stub
(docs, ngrams_path, ngrams_n_size)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/decontamination/decontaminate.py:14
Methodgradient_checkpointing
(self)
hf_olmo/modeling_olmo.py:68
Methodgreedy_until
Generate greedily until a stopping sequence :param requests: list A list of pairs (context, until) context: str
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:97
Methodgreedy_until
(self, requests)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:324
Methodgreedy_until
(self, requests)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/models/textsynth.py:117
Methodgreedy_until
(self, requests)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/models/dummy.py:22
Methodgreedy_until
Generate greedily until a stopping sequence :param requests: list A list of pairs (context, until) context: str
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:97
Methodgreedy_until
(self, requests)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:324
Methodgreedy_until
(self, requests)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/models/textsynth.py:117
Methodgreedy_until
(self, requests)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/models/dummy.py:22
Functionhandle_sigterm
(sig, frame)
inference/efficiency/dependencies/efficiency-pentathlon/eb_gantry/__main__.py:61
Methodhas_instance_conversion
(self, format: InstanceFormat)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/task.py:133
Methodhas_instance_conversion
(self, format: InstanceFormat)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/task.py:133
Methodhas_split
(self, split: str)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/eleuther.py:74
Methodhas_split
(self, split: str)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/eleuther.py:200
Methodhas_split
(self, split: str)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/eleuther.py:233
Methodhas_split
(self, split: str)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/efficiency_benchmark.py:44
Methodhas_split
(self, split: str)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/tasks/huggingface.py:65
Methodhas_split
(self, split: str)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/eleuther.py:74
Methodhas_split
(self, split: str)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/eleuther.py:200
Methodhas_split
(self, split: str)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/eleuther.py:233
Methodhas_split
(self, split: str)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/efficiency_benchmark.py:44
Methodhas_split
(self, split: str)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/huggingface.py:65
Methodhas_test_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/drop.py:51
Methodhas_test_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/sciq.py:35
Methodhas_test_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/cbt.py:42
Methodhas_test_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/triviaqa.py:42
Methodhas_test_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/piqa.py:39
Methodhas_test_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/winogrande.py:44
Methodhas_test_docs
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/hellaswag.py:41
← previousnext →2,401–2,500 of 3,821, ranked by callers