Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ModalityDance/Omni-R1
/ functions
Functions
49,163 in github.com/ModalityDance/Omni-R1
⨍
Functions
49,163
◇
Types & classes
12,513
↳
Endpoints
884
↓ 4,930 callers
Method
to
(self, *args, **kwargs)
src/transformers/src/transformers/modeling_utils.py:2857
↓ 4,680 callers
Method
from_pretrained
r""" Instantiate a [`ModelCard`] from a pre-trained model model card. Parameters: pretrained_model_name_or_path: either:
src/transformers/src/transformers/modelcard.py:125
↓ 4,098 callers
Function
model
r""" # Using torch.hub ! import torch model = torch.hub.load('huggingface/transformers', 'model', 'google-bert/be
src/transformers/hubconf.py:73
↓ 3,498 callers
Method
size
Number of byte required for this data type Returns: Integer > 0
src/transformers/src/transformers/onnx/utils.py:28
↓ 2,349 callers
Function
requires_backends
(obj, backends)
src/transformers/src/transformers/utils/import_utils.py:1551
↓ 2,331 callers
Method
reshape
separate heads
src/transformers/src/transformers/models/distilbert/modeling_distilbert.py:275
↓ 2,171 callers
Method
eval
(**kwargs)
src/transformers/tests/models/auto/test_modeling_flax_auto.py:63
↓ 2,042 callers
Method
info
(self, msg)
src/transformers/tests/trainer/test_trainer_utils.py:482
↓ 1,917 callers
Function
is_torch_available
()
src/transformers/src/transformers/utils/import_utils.py:298
↓ 1,879 callers
Method
pop
(self, *args, **kwargs)
src/transformers/src/transformers/utils/generic.py:422
↓ 1,848 callers
Method
split
Will look for the words added to the trie within `text`. Output is the original string splitted along the boundaries of the words fou
src/transformers/src/transformers/tokenization_utils.py:106
↓ 1,785 callers
Function
ids_tensor
Creates a random int32 tensor of the shape within the vocab size.
src/transformers/tests/generation/test_flax_utils.py:41
↓ 1,735 callers
Method
replace
(match)
src/transformers/utils/diff_model_converter.py:194
↓ 1,716 callers
Method
cat
Concatenates transformations along a new dimension. Args: ts: A list of T objects dim:
src/transformers/src/transformers/models/esm/openfold_utils/rigid_utils.py:1104
↓ 1,698 callers
Method
get
(self, key, default)
src/transformers/src/transformers/models/auto/auto_factory.py:759
↓ 1,541 callers
Method
to
(self, device)
src/transformers/src/transformers/models/detr/modeling_detr.py:2300
↓ 1,266 callers
Method
unsqueeze
Analogous to torch.unsqueeze. The dimension is relative to the shared dimensions of the rotation/translation. Args: dim:
src/transformers/src/transformers/models/esm/openfold_utils/rigid_utils.py:1087
↓ 1,247 callers
Method
keys
Called when a proxy object is has the keys() method called. This is what happens when ** is called on a proxy. This should return an iterator
src/transformers/src/transformers/utils/fx.py:1413
↓ 1,247 callers
Function
shape_list
(*args, **kwargs)
src/transformers/src/transformers/utils/dummy_tf_objects.py:166
↓ 1,096 callers
Method
encode
Uses the `pre_processor` to prepare the inputs for the `model`.
src/transformers/src/transformers/agents/tools.py:565
↓ 1,069 callers
Method
save_pretrained
Save a model card object to the directory or file `save_directory_or_file`.
src/transformers/src/transformers/modelcard.py:113
↓ 1,021 callers
Method
items
(self)
src/transformers/src/transformers/models/auto/auto_factory.py:776
↓ 998 callers
Method
items
(self)
src/transformers/src/transformers/feature_extraction_utils.py:113
↓ 996 callers
Method
post_init
r""" Safety checker that arguments are correct - also replaces some NoneType arguments with their default values.
src/transformers/src/transformers/utils/quantization_config.py:256
↓ 990 callers
Function
tokenizer
r""" # Using torch.hub ! import torch tokenizer = torch.hub.load('huggingface/transformers', 'tokenizer', 'google-bert/bert-b
src/transformers/hubconf.py:59
↓ 932 callers
Method
generate
(model: Phi3ForCausalLM, prompt_tokens: torch.LongTensor, max_seq_len: int)
src/transformers/tests/models/phi3/test_modeling_phi3.py:73
↓ 783 callers
Method
normal_
(self, data: torch.Tensor)
src/transformers/src/transformers/models/deprecated/graphormer/modeling_graphormer.py:711
↓ 737 callers
Method
decode
Uses the `post_processor` to decode the model output.
src/transformers/src/transformers/agents/tools.py:578
↓ 657 callers
Method
write
(self, msg)
src/transformers/scripts/benchmark/trainer-benchmark.py:139
↓ 596 callers
Method
float
(self, *args)
src/transformers/src/transformers/modeling_utils.py:2896
↓ 592 callers
Method
load
Loads the dictionary from a text file with the format: ``` <symbol0> <count0> <symbol1> <count1> ... ```
src/transformers/src/transformers/models/biogpt/convert_biogpt_original_pytorch_checkpoint_to_pytorch.py:78
↓ 539 callers
Function
get_initializer
Creates a `keras.initializers.TruncatedNormal` with the given range. Args: initializer_range (*float*, defaults to 0.02): Standard d
src/transformers/src/transformers/modeling_tf_utils.py:3545
↓ 538 callers
Function
is_vision_available
()
src/transformers/src/transformers/utils/import_utils.py:970
↓ 533 callers
Method
batch_decode
This method forwards all its arguments to BertTokenizerFast's [`~PreTrainedTokenizer.batch_decode`]. Please refer to the docstring of
src/transformers/src/transformers/models/tvp/processing_tvp.py:111
↓ 525 callers
Function
is_tf_available
()
src/transformers/src/transformers/utils/import_utils.py:554
↓ 520 callers
Method
from_pretrained
(cls, pretrained_model_name_or_path: str, **kwargs)
src/transformers/examples/research_projects/lxmert/utils.py:177
↓ 482 callers
Method
open
(*args, **kwargs)
src/transformers/tests/pipelines/test_pipelines_image_to_text.py:38
↓ 473 callers
Method
format
(self, record)
src/transformers/src/transformers/agents/agents.py:76
↓ 445 callers
Method
detach
Returns a copy of the Rotation whose underlying Tensor has been detached from its torch graph. Returns: A copy of the Ro
src/transformers/src/transformers/models/esm/openfold_utils/rigid_utils.py:710
↓ 422 callers
Method
expand_dims
Expands 2-dimensional `image` to 3 dimensions. Args: image (`PIL.Image.Image` or `np.ndarray` or `torch.Tensor`):
src/transformers/src/transformers/image_utils.py:525
↓ 380 callers
Method
keys
(self)
src/transformers/src/transformers/models/auto/auto_factory.py:751
↓ 373 callers
Function
prune_linear_layer
Prune a linear layer to keep only entries in index. Used to remove heads. Args: layer (`torch.nn.Linear`): The layer to prune.
src/transformers/src/transformers/pytorch_utils.py:51
↓ 368 callers
Method
tokenize
Tokenize the provided input and eventually returns corresponding tokens id: - **text_input**: String to tokenize - **return_ids**: Bo
src/transformers/src/transformers/commands/serving.py:179
↓ 360 callers
Method
update
Updates the cache with the new `key_states` and `value_states` for the layer `layer_idx`. Parameters: key_states (`torch
src/transformers/src/transformers/cache_utils.py:34
↓ 357 callers
Method
mean
Returns the mean of the distribution.
src/transformers/src/transformers/time_series_utils.py:43
↓ 353 callers
Method
encoder
(self)
src/transformers/src/transformers/models/whisper/modeling_tf_whisper.py:1250
↓ 349 callers
Method
classifier
(self, x: tf.Tensor)
src/transformers/src/transformers/models/resnet/modeling_tf_resnet.py:540
↓ 340 callers
Method
convert_tokens_to_ids
Converts a token string (or a sequence of tokens) in a single integer id (or a sequence of ids), using the vocabulary. Args:
src/transformers/src/transformers/tokenization_utils.py:708
↓ 336 callers
Method
convert_to_tensor
(self, symbols)
src/transformers/src/transformers/models/deprecated/transfo_xl/tokenization_transfo_xl.py:482
↓ 329 callers
Method
update
( self, key_states: torch.Tensor, value_states: torch.Tensor, layer_idx: int,
src/transformers/src/transformers/models/jamba/modeling_jamba.py:238
↓ 326 callers
Method
encode_plus
Prepare a table and a string for the model. Args: table (`pd.DataFrame`): Table containing tabular data.
src/transformers/src/transformers/models/tapas/tokenization_tapas.py:898
↓ 317 callers
Function
floats_tensor
Creates a random float32 tensor
src/transformers/tests/test_modeling_common.py:4854
↓ 316 callers
Method
values
(self)
src/transformers/src/transformers/feature_extraction_utils.py:109
↓ 306 callers
Method
run_common_tests
(self)
src/transformers/tests/models/pvt/test_modeling_pvt.py:46
↓ 283 callers
Method
train
Main training entry point. Args: resume_from_checkpoint (`str` or `bool`, *optional*): If a `str`, local
src/transformers/src/transformers/trainer.py:1863
↓ 282 callers
Function
pipeline
Utility factory method to build a [`Pipeline`]. Pipelines are made of: - A [tokenizer](tokenizer) in charge of mapping raw textual
src/transformers/src/transformers/pipelines/__init__.py:552
↓ 281 callers
Function
is_tokenizers_available
()
src/transformers/src/transformers/utils/import_utils.py:965
↓ 280 callers
Method
log
Log `logs` on the various objects watching training. Subclass and override this method to inject custom behavior. Args:
src/transformers/src/transformers/trainer.py:3265
↓ 276 callers
Method
mean
(self)
src/transformers/src/transformers/models/esm/modeling_esmfold.py:1270
↓ 267 callers
Method
to_tuple
Convert self to a tuple containing all the attributes/keys that are not `None`.
src/transformers/src/transformers/utils/generic.py:454
↓ 261 callers
Method
batch_encode_plus
Prepare a table and a list of strings for the model. <Tip warning={true}> This method is deprecated, `__call__` should be u
src/transformers/src/transformers/models/tapas/tokenization_tapas.py:617
↓ 259 callers
Method
get_vocab
(self)
src/transformers/src/transformers/models/t5/tokenization_t5.py:245
↓ 258 callers
Method
parse
Overwrites the `parse` method to incorporate a skip for CUDA tests, and remove logs and dataset prints before calling `super().parse`
src/transformers/src/transformers/testing_utils.py:2356
↓ 248 callers
Method
pad
Pad a single encoded input or a batch of encoded inputs up to predefined length or to the max sequence length in the batch. Padding s
src/transformers/src/transformers/models/luke/tokenization_luke.py:1372
↓ 244 callers
Method
save_pretrained
(self, save_directory)
src/transformers/src/transformers/models/rag/retrieval_rag.py:459
↓ 227 callers
Method
read
(self, n=-1)
src/transformers/src/transformers/commands/lfs.py:142
↓ 225 callers
Function
nested_simplify
Simplifies an object by rounding float numbers, and downcasting tensors/numpy arrays to get simple equality test within tests.
src/transformers/src/transformers/testing_utils.py:2086
↓ 224 callers
Method
model
(self)
src/transformers/tests/models/t5/test_modeling_t5.py:1132
↓ 220 callers
Function
is_flax_available
()
src/transformers/src/transformers/utils/import_utils.py:574
↓ 219 callers
Method
to_dict
Serializes this instance to a Python dictionary.
src/transformers/src/transformers/modelcard.py:237
↓ 216 callers
Function
get_size_dict
Converts the old size parameter in the config into the new dict expected in the config. This is to ensure backwards compatibility with the ol
src/transformers/src/transformers/image_processing_utils.py:208
↓ 210 callers
Method
copy
Creates a new instance of this constraint. Args: stateful(`bool`): Whether to not only copy the constraint for new insta
src/transformers/src/transformers/generation/beam_constraints.py:114
↓ 209 callers
Method
decoder
(self)
src/transformers/src/transformers/models/whisper/modeling_tf_whisper.py:1247
↓ 207 callers
Method
compile
This is a thin wrapper that sets the model's loss output head as the loss if the user does not specify a loss function themselves.
src/transformers/src/transformers/modeling_tf_utils.py:1531
↓ 206 callers
Method
attention
(self, hidden_state: torch.Tensor, attention_mask: torch.Tensor)
src/transformers/src/transformers/models/bridgetower/modeling_bridgetower.py:203
↓ 206 callers
Method
map
Map inputs to attention experts according to routing decision and compute query projection inside each experts.
src/transformers/src/transformers/models/jetmoe/modeling_jetmoe.py:363
↓ 206 callers
Method
output
(result: Union[str, int])
src/transformers/src/transformers/models/whisper/english_normalizer.py:222
↓ 192 callers
Function
random_attention_mask
(shape, rng=None)
src/transformers/tests/generation/test_flax_utils.py:59
↓ 192 callers
Method
save
Saves the relevant code files for your tool so it can be pushed to the Hub. This will copy the code of your tool in `output_dir` as w
src/transformers/src/transformers/agents/tools.py:144
↓ 188 callers
Function
flatten_dict
Flatten a nested dict into a single level dict.
src/transformers/src/transformers/utils/generic.py:587
↓ 173 callers
Method
apply
Applies the transformation to a coordinate tensor. Args: pts: A [*, 3] coordinate tensor. Returns: T
src/transformers/src/transformers/models/esm/openfold_utils/rigid_utils.py:938
↓ 173 callers
Method
to
(self, device)
src/transformers/src/transformers/models/grounding_dino/modeling_grounding_dino.py:2625
↓ 171 callers
Method
push_to_hub
Upload the tool to the Hub. For this method to work properly, your tool must have been defined in a separate module (not `__main__`)
src/transformers/src/transformers/agents/tools.py:316
↓ 167 callers
Method
convert_ids_to_tokens
(self, ids: int, skip_special_tokens: bool = False)
src/transformers/src/transformers/tokenization_utils.py:1028
↓ 166 callers
Method
get_input_embeddings
(self)
src/transformers/src/transformers/models/t5/modeling_t5.py:963
↓ 160 callers
Method
backward
(ctx, grad_output)
src/transformers/src/transformers/models/sew_d/modeling_sew_d.py:534
↓ 159 callers
Method
to_json_string
Serializes this instance to a JSON string.
src/transformers/src/transformers/modelcard.py:242
↓ 157 callers
Method
copy
(self)
src/transformers/src/transformers/models/oneformer/convert_to_hf_oneformer.py:87
↓ 154 callers
Method
prepare_config_and_inputs_for_common
(self)
src/transformers/tests/models/t5/test_modeling_t5.py:530
↓ 153 callers
Function
is_sentencepiece_available
()
src/transformers/src/transformers/utils/import_utils.py:907
↓ 149 callers
Function
get_values
(model_mapping)
src/transformers/src/transformers/models/auto/auto_factory.py:677
↓ 141 callers
Method
time
(self)
src/transformers/utils/notification_service.py:165
↓ 136 callers
Function
get_regression_trainer
( a=0, b=0, double_output=False, train_len=64, eval_len=64, pretrained=True, keep_report_to=False, **k
src/transformers/tests/trainer/test_trainer.py:430
↓ 133 callers
Method
from_dict
Constructs a `ModelCard` from a Python dictionary of parameters.
src/transformers/src/transformers/modelcard.py:219
↓ 132 callers
Function
get_tf_activation
(activation_string)
src/transformers/src/transformers/activations_tf.py:143
↓ 132 callers
Method
values
(self)
src/transformers/src/transformers/models/auto/auto_factory.py:768
↓ 128 callers
Method
is_available
()
src/transformers/src/transformers/hyperparameter_search.py:111
↓ 125 callers
Function
get_image_size
Returns the (height, width) dimensions of the image. Args: image (`np.ndarray`): The image to get the dimensions of.
src/transformers/src/transformers/image_utils.py:281
↓ 125 callers
Function
set_seed
(args)
src/transformers/examples/legacy/run_swag.py:224
↓ 124 callers
Method
detach
Need to document this
src/transformers/examples/research_projects/bertabs/modeling_bertabs.py:569
next →
1–100 of 49,163, ranked by callers