MCPcopy Create free account

hub / github.com/allenai/OLMo / functions

Functions3,821 in github.com/allenai/OLMo

Method__init__
(self, arr, fn)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/utils.py:123
Method__init__
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:26
Method__init__
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:379
Method__init__
(self, cachinglm)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:694
Method__init__
LM wrapper that returns cached results if they exist, and uses the underlying LM if not. :param lm: LM Underlying LM :par
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:709
Method__init__
(self, request_type, args, index=None)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/base.py:776
Method__init__
(self, f, n)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/metrics.py:193
Method__init__
( self, ngram_n=13, window_to_remove=200, too_dirty_cutoff=10, minimum
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/decontamination/janitor.py:103
Method__init__
(self, file_path, compression_level=3)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/decontamination/archiver.py:23
Method__init__
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/decontamination/archiver.py:45
Method__init__
(self, file_path, mode="rb+")
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/decontamination/archiver.py:73
Method__init__
(self, file_path)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/decontamination/archiver.py:93
Method__init__
(self, file)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/decontamination/archiver.py:150
Method__init__
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/truthfulqa.py:155
Method__init__
(self, subject)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_test.py:109
Method__init__
(self, sacrebleu_dataset, sacrebleu_language_pair=None)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/translation.py:98
Method__init__
SAT Analog Questions is not publicly available. You must request the data by emailing Peter Turney and then download it to a local di
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/sat.py:34
Method__init__
StoryCloze is not publicly available. You must download the data by following https://cs.rochester.edu/nlp/rocstories/ and pass the f
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/storycloze.py:40
Method__init__
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/headqa.py:83
Method__init__
(self, f)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/tasks/race.py:30
Method__init__
BuilderConfig for Wikitext Args: data_url: `string`, url to the dataset (word or raw level) **kwargs: keyword arguments fo
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/datasets/wikitext/wikitext.py:48
Method__init__
BuilderConfig for Hendrycks ETHICS. Args: prefix: *string*, prefix to add to the dataset name for path location. features: *l
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/datasets/hendrycks_ethics/hendrycks_ethics.py:52
Method__init__
BuilderConfig for GPT3 Arithmetic dataset. Args: url: *string*, the url to the specific subset of the GPT3 Arithmetic dataset.
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/datasets/arithmetic/arithmetic.py:49
Method__init__
( self, device="cuda", pretrained="gpt2", revision="main", subfolder=N
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/models/gpt2.py:7
Method__init__
:param engine: str TextSynth API engine (e.g. `gptj_6B`) :param truncate: bool Truncate input if too long (if
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/models/textsynth.py:42
Method__init__
(self)
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/models/dummy.py:7
Method__init__
:param engine: str OpenAI API engine (e.g. davinci) :param truncate: bool Truncate input if too long (if Fal
inference/efficiency/dependencies/efficiency-pentathlon/efficiency_benchmark/dependencies/lm_eval/models/gpt3.py:61
Method__init__
(self, pretrained_model_dir)
inference/efficiency/dependencies/previous_version/olmo_efficiency.py:40
Method__init__
(self, pretrained_model_dir)
inference/efficiency/dependencies/previous_version/olmo_efficiency.py:68
Method__init__
(self, *, version_override: Optional[str] = None)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/task.py:96
Method__init__
(self, answer_options: Sequence[str])
inference/efficiency/dependencies/previous_version/efficiency_benchmark/task.py:189
Method__init__
( self, *, cmd: List[str], task: str, scenario: str, max_batch
inference/efficiency/dependencies/previous_version/efficiency_benchmark/steps.py:21
Method__init__
(self, task: Union[str, Task])
inference/efficiency/dependencies/previous_version/efficiency_benchmark/steps.py:200
Method__init__
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/steps.py:267
Method__init__
(self, task: Union[str, Task], output_file: Optional[str] = None)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/steps.py:296
Method__init__
(self, fn: Callable, inner_sequence: Sequence)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/utils.py:18
Method__init__
binary_cmd: the command to start the inference binary
inference/efficiency/dependencies/previous_version/efficiency_benchmark/stdio_wrapper.py:17
Method__init__
( self, interval: float = 0.1, # gpu_ids: Optional[Iterable[int]] = None, **kw
inference/efficiency/dependencies/previous_version/efficiency_benchmark/efficiency/profiler.py:19
Method__init__
(self, stop_event, *args, **kwargs)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/efficiency/power_monitor.py:17
Method__init__
(self, inner_sequence: Sequence, indices: Optional[Sequence[int]] = None)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tango_utils/sequences.py:52
Method__init__
(self, inner_sequence: Sequence, s: slice)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tango_utils/sequences.py:110
Method__init__
(self, *sequences: Sequence)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tango_utils/sequences.py:149
Method__init__
(self, fn: Callable, inner_sequence: Sequence)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tango_utils/sequences.py:211
Method__init__
(self, filename: Union[str, PathLike], read_only: bool = False)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tango_utils/sequences.py:280
Method__init__
(self, buffer: io.BytesIO)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tango_utils/det_hash.py:101
Method__init__
( self, nan_strategy: Union[str, float] = "warn", **kwargs: Dict[str, Any], )
inference/efficiency/dependencies/previous_version/efficiency_benchmark/metrics/perplexity.py:8
Method__init__
(self, num_classes: int)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/metrics/accuracy.py:31
Method__init__
( self, base: int = 2, # Does anyone ever use anything but 2 here? nan_strategy: Unio
inference/efficiency/dependencies/previous_version/efficiency_benchmark/metrics/bleu.py:9
Method__init__
( self, base: int = 2, # Does anyone ever use anything but 2 here? nan_strategy: Unio
inference/efficiency/dependencies/previous_version/efficiency_benchmark/metrics/entropy.py:9
Method__init__
(self, dataset_name: str, *, version_override: Optional[str] = None)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/metaicl.py:12
Method__init__
( self, dataset_name: str, *, version_override: Optional[str] = None, )
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/p3.py:8
Method__init__
( self, eleuther_task: Union[str, Callable[[], EAITask]], *, answer_options: S
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/eleuther.py:148
Method__init__
(self, *, version_override: Optional[str] = None)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/eleuther.py:195
Method__init__
( self, eleuther_task: Union[str, Callable[[], EAITask]], *, version_override:
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/eleuther.py:222
Method__init__
( self, eleuther_task: Union[str, Callable[[], EAITask]], *, answer_options: S
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/eleuther.py:257
Method__init__
( self, dataset_path: str, dataset_name: Optional[str] = None, )
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/efficiency_benchmark.py:148
Method__init__
(self, subset: str)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/efficiency_benchmark.py:170
Method__init__
(self, subset: str, number_of_classes: int = 2)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/raft.py:40
Method__init__
( self, dataset_path: str, dataset_name: Optional[str] = None, *, version_override: Optional[str] = No
inference/efficiency/dependencies/previous_version/efficiency_benchmark/tasks/huggingface.py:56
Method__init__
(self, arr, fn)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/utils.py:123
Method__init__
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:26
Method__init__
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:379
Method__init__
(self, cachinglm)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:694
Method__init__
LM wrapper that returns cached results if they exist, and uses the underlying LM if not. :param lm: LM Underlying LM :par
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:709
Method__init__
(self, request_type, args, index=None)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/base.py:776
Method__init__
(self, f, n)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/metrics.py:193
Method__init__
( self, ngram_n=13, window_to_remove=200, too_dirty_cutoff=10, minimum
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/decontamination/janitor.py:103
Method__init__
(self, file_path, compression_level=3)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/decontamination/archiver.py:23
Method__init__
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/decontamination/archiver.py:45
Method__init__
(self, file_path, mode="rb+")
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/decontamination/archiver.py:73
Method__init__
(self, file_path)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/decontamination/archiver.py:93
Method__init__
(self, file)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/decontamination/archiver.py:150
Method__init__
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/truthfulqa.py:155
Method__init__
(self, subject)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/hendrycks_test.py:109
Method__init__
(self, sacrebleu_dataset, sacrebleu_language_pair=None)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/translation.py:98
Method__init__
SAT Analog Questions is not publicly available. You must request the data by emailing Peter Turney and then download it to a local di
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/sat.py:34
Method__init__
StoryCloze is not publicly available. You must download the data by following https://cs.rochester.edu/nlp/rocstories/ and pass the f
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/storycloze.py:40
Method__init__
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/headqa.py:83
Method__init__
(self, f)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/tasks/race.py:30
Method__init__
BuilderConfig for Wikitext Args: data_url: `string`, url to the dataset (word or raw level) **kwargs: keyword arguments fo
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/datasets/wikitext/wikitext.py:48
Method__init__
BuilderConfig for Hendrycks ETHICS. Args: prefix: *string*, prefix to add to the dataset name for path location. features: *l
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/datasets/hendrycks_ethics/hendrycks_ethics.py:52
Method__init__
BuilderConfig for GPT3 Arithmetic dataset. Args: url: *string*, the url to the specific subset of the GPT3 Arithmetic dataset.
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/datasets/arithmetic/arithmetic.py:49
Method__init__
( self, device="cuda", pretrained="gpt2", revision="main", subfolder=N
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/models/gpt2.py:7
Method__init__
:param engine: str TextSynth API engine (e.g. `gptj_6B`) :param truncate: bool Truncate input if too long (if
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/models/textsynth.py:42
Method__init__
(self)
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/models/dummy.py:7
Method__init__
:param engine: str OpenAI API engine (e.g. davinci) :param truncate: bool Truncate input if too long (if Fal
inference/efficiency/dependencies/previous_version/efficiency_benchmark/dependencies/lm_eval/models/gpt3.py:61
Method__init__
(self, stop_id_sequences)
inference/eval/utils.py:18
Method__init__
(self, m, device)
inference/compression/dependencies/AutoGPTQ/auto_gptq/modeling/_base.py:243
Method__init__
(self)
inference/compression/dependencies/AutoGPTQ/auto_gptq/modeling/auto.py:41
Method__init__
Calculate perplexity using the same method as seen in llama.cpp. Parameters ---------- model : AutoModelForCausalLM
inference/compression/dependencies/AutoGPTQ/auto_gptq/utils/perplexity_utils.py:15
Method__init__
( self, adapter_name: str, linear_module: torch.nn.Linear, r: int = 0,
inference/compression/dependencies/AutoGPTQ/auto_gptq/utils/peft_utils.py:25
Method__init__
(self, shape=1)
inference/compression/dependencies/AutoGPTQ/auto_gptq/quantization/quantizer.py:17
Method__init__
(self, layer)
inference/compression/dependencies/AutoGPTQ/auto_gptq/quantization/gptq.py:19
Method__init__
( self, gate_proj, down_proj, up_proj, )
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/fused_llama_mlp.py:200
Method__init__
(self, config)
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/fused_gptj_attn.py:48
Method__init__
( self, hidden_size, num_heads, qkv_proj, o_proj, rotary_emb,
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/fused_llama_attn.py:18
Method__init__
( self, bits, group_size, infeatures, outfeatures, bias,
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/qlinear/qlinear_cuda_old.py:25
Method__init__
(self, quant_linear_module)
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/qlinear/__init__.py:5
Method__init__
(self, bits, group_size, infeatures, outfeatures, bias, trainable=False)
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/qlinear/qlinear_triton.py:29
Method__init__
( self, bits, group_size, infeatures, outfeatures, bias, kernel_switch_threshold=128, trainable=False
inference/compression/dependencies/AutoGPTQ/auto_gptq/nn_modules/qlinear/qlinear_cuda.py:26
← previousnext →1,201–1,300 of 3,821, ranked by callers