MCPcopy Create free account

hub / github.com/ModalityDance/Omni-R1 / functions

Functions49,163 in github.com/ModalityDance/Omni-R1

↓ 14 callersMethod__init__
(self, input_dim: int, config: SegformerConfig, **kwargs)
src/transformers/src/transformers/models/segformer/modeling_tf_segformer.py:820
↓ 14 callersMethod__init__
(self, config: AutoformerConfig)
src/transformers/src/transformers/models/autoformer/modeling_autoformer.py:1428
↓ 14 callersMethod__init__
(self, config: LayoutLMv3Config, **kwargs)
src/transformers/src/transformers/models/layoutlmv3/modeling_tf_layoutlmv3.py:559
↓ 14 callersMethod__init__
(self, config: VideoMAEConfig)
src/transformers/src/transformers/models/videomae/modeling_videomae.py:409
↓ 14 callersMethod__init__
Construct a layernorm module in the T5 style. No bias and no subtraction of mean.
src/transformers/src/transformers/models/pix2struct/modeling_pix2struct.py:54
↓ 14 callersMethod__init__
( self, num_heads, embed_dim, kernel_size, padding_q, padding_
src/transformers/src/transformers/models/cvt/modeling_cvt.py:374
↓ 14 callersMethod__init__
(self, config, dim, num_heads, drop_path_rate=0.0)
src/transformers/src/transformers/models/deprecated/nat/modeling_nat.py:431
↓ 14 callersMethod__init__
(self, config: MBartConfig)
src/transformers/src/transformers/models/mbart/modeling_mbart.py:1270
↓ 14 callersMethod__init__
(self, cfg, input_shape: Dict[str, ShapeSpec])
src/transformers/examples/research_projects/visual_bert/modeling_frcnn.py:1502
↓ 14 callersMethod__init__
(self, cfg, input_shape: Dict[str, ShapeSpec])
src/transformers/examples/research_projects/lxmert/modeling_frcnn.py:1502
↓ 14 callersFunction_get_library_root_logger
()
src/transformers/src/transformers/utils/logging.py:77
↓ 14 callersMethod_shape
(self, tensor: torch.Tensor, seq_len: int, bsz: int)
src/transformers/src/transformers/models/hubert/modeling_hubert.py:429
↓ 14 callersMethod_shape
(self, tensor: torch.Tensor, seq_len: int, bsz: int)
src/transformers/src/transformers/models/bart/modeling_bart.py:165
↓ 14 callersMethod_shape
(self, tensor: torch.Tensor, seq_len: int, bsz: int)
src/transformers/src/transformers/models/data2vec/modeling_data2vec_audio.py:359
↓ 14 callersMethod_shape
(self, tensor: torch.Tensor, seq_len: int, bsz: int)
src/transformers/src/transformers/models/musicgen_melody/modeling_musicgen_melody.py:206
↓ 14 callersMethod_shape
(self, tensor: torch.Tensor, seq_len: int, bsz: int)
src/transformers/src/transformers/models/unispeech/modeling_unispeech.py:465
↓ 14 callersMethod_shape
(self, tensor: torch.Tensor, seq_len: int, bsz: int)
src/transformers/src/transformers/models/wav2vec2/modeling_wav2vec2.py:529
↓ 14 callersMethod_shape
(self, tensor: torch.Tensor, seq_len: int, bsz: int)
src/transformers/src/transformers/models/musicgen/modeling_musicgen.py:190
↓ 14 callersMethod_shape
(self, tensor: torch.Tensor, seq_len: int, bsz: int)
src/transformers/src/transformers/models/unispeech_sat/modeling_unispeech_sat.py:482
↓ 14 callersMethod_shape
(self, tensor: torch.Tensor, seq_len: int, bsz: int)
src/transformers/src/transformers/models/sew/modeling_sew.py:429
↓ 14 callersMethodaggregate
(self, pre_entities: List[dict], aggregation_strategy: AggregationStrategy)
src/transformers/src/transformers/pipelines/token_classification.py:417
↓ 14 callersMethodbuild
(self, input_shape=None)
src/transformers/src/transformers/models/openai/modeling_tf_openai.py:182
↓ 14 callersMethodcall_event
(self, event, args, state, control, **kwargs)
src/transformers/src/transformers/trainer_callback.py:505
↓ 14 callersFunctionfloats_list
Creates a random float32 tensor
src/transformers/tests/models/speecht5/test_feature_extraction_speecht5.py:38
↓ 14 callersMethodfrom_pretrained
(cls, pretrained_model_name_or_path, **kwargs)
src/transformers/src/transformers/models/wav2vec2/processing_wav2vec2.py:51
↓ 14 callersMethodfrom_pretrained
(cls, *args, **kwargs)
src/transformers/src/transformers/models/vision_text_dual_encoder/modeling_vision_text_dual_encoder.py:409
↓ 14 callersMethodget_bias
Dict of bias attached to an LM head. The key represents the name of the bias attribute. Return: `tf.Variable`: The weigh
src/transformers/src/transformers/modeling_tf_utils.py:1966
↓ 14 callersMethodget_lang_id
(self, lang: str)
src/transformers/src/transformers/models/m2m_100/tokenization_m2m_100.py:361
↓ 14 callersMethodget_metadata
(self, labels, start, total_length, offset, get_indices=False)
src/transformers/src/transformers/models/deprecated/jukebox/modeling_jukebox.py:1905
↓ 14 callersMethodget_words_and_boxes_batch
(self)
src/transformers/tests/models/layoutlmv2/test_tokenization_layoutlmv2.py:78
↓ 14 callersMethodget_words_and_boxes_batch
(self)
src/transformers/tests/models/layoutxlm/test_tokenization_layoutxlm.py:72
↓ 14 callersMethodget_words_and_boxes_batch
(self)
src/transformers/tests/models/udop/test_tokenization_udop.py:71
↓ 14 callersFunctionis_speech_available
()
src/transformers/src/transformers/utils/import_utils.py:1100
↓ 14 callersFunctionload_image
Loads `image` to a PIL Image. Args: image (`str` or `PIL.Image.Image`): The image to convert to the PIL Image format.
src/transformers/src/transformers/image_utils.py:344
↓ 14 callersFunctionmake_divisible
Ensure that all layers have a channel count that is divisible by `divisor`. This function is taken from the original TensorFlow repo. It can
src/transformers/src/transformers/models/mobilevitv2/modeling_mobilevitv2.py:60
↓ 14 callersMethodnormalize
Normalizes `image` with `mean` and `std`. Note that this will trigger a conversion of `image` to a NumPy array if it's a PIL Image.
src/transformers/src/transformers/image_utils.py:545
↓ 14 callersMethodpad
Pads a batch of images to the bottom and right of the image with zeros to the size of largest height and width in the batch and optio
src/transformers/src/transformers/models/bridgetower/image_processing_bridgetower.py:318
↓ 14 callersMethodpad
Pads a batch of images to the bottom and right of the image with zeros to the size of largest height and width in the batch and optio
src/transformers/src/transformers/models/vilt/image_processing_vilt.py:282
↓ 14 callersFunctionparse_commit_message
Parses the commit message to detect if a command is there to skip, force all or part of the CI. Args: commit_message (`str`): The co
src/transformers/utils/tests_fetcher.py:1192
↓ 14 callersMethodprepare_config_and_inputs
(self)
src/transformers/tests/models/hubert/test_modeling_hubert.py:115
↓ 14 callersMethodprepare_config_and_inputs
(self)
src/transformers/tests/models/encoder_decoder/test_modeling_tf_encoder_decoder.py:68
↓ 14 callersMethodprepare_config_and_inputs
(self)
src/transformers/tests/models/pop2piano/test_modeling_pop2piano.py:98
↓ 14 callersMethodreset
(self)
src/transformers/src/transformers/cache_utils.py:1655
↓ 14 callersMethodsave_pretrained
(self, save_directory)
src/transformers/src/transformers/models/deprecated/realm/retrieval_realm.py:123
↓ 14 callersFunctionset_param
(torch_layer, weight, bias=None)
src/transformers/src/transformers/models/reformer/convert_reformer_trax_checkpoint_to_pytorch.py:31
↓ 14 callersFunctiontranspose
Framework-agnostic version of `numpy.transpose` that will work on torch/TensorFlow/Jax tensors as well as NumPy arrays.
src/transformers/src/transformers/utils/generic.py:610
↓ 14 callersFunctionverify_backbone_config_arguments
Verify that the config arguments to be passed to load_backbone are valid
src/transformers/src/transformers/utils/backbone_utils.py:360
↓ 13 callersMethod__init__
(self, config: SwiftFormerConfig, in_features: int, **kwargs)
src/transformers/src/transformers/models/swiftformer/modeling_tf_swiftformer.py:252
↓ 13 callersMethod__init__
(self, config: AlbertConfig)
src/transformers/src/transformers/models/albert/modeling_albert.py:362
↓ 13 callersMethod__init__
(self, config)
src/transformers/src/transformers/models/layoutlmv2/modeling_layoutlmv2.py:271
↓ 13 callersMethod__init__
(self, config: ASTConfig)
src/transformers/src/transformers/models/audio_spectrogram_transformer/modeling_audio_spectrogram_transformer.py:313
↓ 13 callersMethod__init__
(self, config)
src/transformers/src/transformers/models/clipseg/modeling_clipseg.py:344
↓ 13 callersMethod__init__
(self, config)
src/transformers/src/transformers/models/owlvit/modeling_owlvit.py:440
↓ 13 callersMethod__init__
(self, config: CLIPConfig, **kwargs)
src/transformers/src/transformers/models/clip/modeling_tf_clip.py:374
↓ 13 callersMethod__init__
(self, config)
src/transformers/src/transformers/models/siglip/modeling_siglip.py:599
↓ 13 callersMethod__init__
(self, config, **kwargs)
src/transformers/src/transformers/models/blip/modeling_tf_blip_text.py:431
↓ 13 callersMethod__init__
(self, config, layer_num)
src/transformers/src/transformers/models/blip/modeling_blip_text.py:321
↓ 13 callersMethod__init__
(self, config: SegformerConfig, input_dim)
src/transformers/src/transformers/models/segformer/modeling_segformer.py:665
↓ 13 callersMethod__init__
(self, config: MvpConfig)
src/transformers/src/transformers/models/mvp/modeling_mvp.py:1221
↓ 13 callersMethod__init__
(self, config: ClvpDecoderConfig)
src/transformers/src/transformers/models/clvp/modeling_clvp.py:1211
↓ 13 callersMethod__init__
(self, config)
src/transformers/src/transformers/models/layoutlmv3/modeling_layoutlmv3.py:515
↓ 13 callersMethod__init__
( self, config, in_channels, out_channels, stride, dilation,
src/transformers/src/transformers/models/bit/modeling_bit.py:510
↓ 13 callersMethod__init__
(self, config: ViTMAEConfig)
src/transformers/src/transformers/models/vit_mae/modeling_vit_mae.py:553
↓ 13 callersMethod__init__
( self, config, idx, hidden_sizes, key_dim, depths, nu
src/transformers/src/transformers/models/levit/modeling_levit.py:356
↓ 13 callersMethod__init__
(self, config)
src/transformers/src/transformers/models/dinov2/modeling_dinov2.py:332
↓ 13 callersMethod__init__
(self, config, dim, input_resolution, num_heads, shift_size=0)
src/transformers/src/transformers/models/maskformer/modeling_maskformer_swin.py:512
↓ 13 callersMethod__init__
(self, config: ViTHybridConfig)
src/transformers/src/transformers/models/deprecated/vit_hybrid/modeling_vit_hybrid.py:382
↓ 13 callersMethod__init__
(self, config)
src/transformers/src/transformers/models/owlv2/modeling_owlv2.py:448
↓ 13 callersMethod__init__
(self, config)
src/transformers/examples/research_projects/movement-pruning/emmental/modeling_bert_masked.py:288
↓ 13 callersMethod_check_lora_correctly_converted
Utility method to check if the model has correctly adapters injected on it.
src/transformers/tests/peft_integration/test_peft_integration.py:52
↓ 13 callersMethod_check_outputs
(self, output, input_ids, config, use_cache=False, num_return_sequences=1)
src/transformers/tests/generation/test_utils.py:1904
↓ 13 callersFunction_compute_mask_indices
Computes random mask spans for a given shape. Used to implement [SpecAugment: A Simple Data Augmentation Method for ASR](https://arxiv.org/ab
src/transformers/src/transformers/models/wav2vec2/modeling_wav2vec2.py:139
↓ 13 callersMethod_get_logits_processor_and_warper_kwargs
( input_length, forced_bos_token_id=None, forced_eos_token_id=None, )
src/transformers/tests/generation/test_utils.py:122
↓ 13 callersMethod_get_tensors
(self, length)
src/transformers/tests/generation/test_stopping_criteria.py:41
↓ 13 callersMethod_shape
(self, tensor: torch.Tensor, seq_len: int, bsz: int)
src/transformers/src/transformers/models/opt/modeling_opt.py:119
↓ 13 callersMethod_split_seq_length_dim_to
splits sequence length dim of vectors into `dim_factor_1` and `dim_factor_2` dims
src/transformers/src/transformers/models/reformer/modeling_reformer.py:308
↓ 13 callersMethodaccumulate
(grad1, grad2)
src/transformers/tests/optimization/test_optimization_tf.py:74
↓ 13 callersMethodadd_arrays
Add `arrays` to the internal storage, Will initialize the storage to the full size at the first arrays passed so that if we're bound
src/transformers/src/transformers/trainer_pt_utils.py:496
↓ 13 callersFunctionassert_device_map
(device_map, num_blocks)
src/transformers/src/transformers/utils/model_parallel_utils.py:19
↓ 13 callersMethodassert_outputs_same
(self, after_outputs, outputs)
src/transformers/tests/test_modeling_tf_common.py:410
↓ 13 callersMethodbuild
(self, input_shape=None)
src/transformers/src/transformers/models/idefics/vision_tf.py:311
↓ 13 callersFunctionconvert_to_rgb
(image)
src/transformers/src/transformers/models/idefics/image_processing_idefics.py:38
↓ 13 callersMethodconvert_to_tensors
Convert the inner content to tensors. Args: tensor_type (`str` or [`~utils.TensorType`], *optional*): Th
src/transformers/src/transformers/tokenization_utils_base.py:684
↓ 13 callersFunctioncopy_func
Returns a copy of a function f.
src/transformers/src/transformers/utils/doc.py:1184
↓ 13 callersFunctioncreate_optimizer
Creates an optimizer with a learning rate schedule using a warmup phase followed by a linear decay. Args: init_lr (`float`):
src/transformers/src/transformers/optimization_tf.py:96
↓ 13 callersMethodfrom_pretrained
r""" Example: ```python >>> from transformers import EncoderDecoderModel >>> model = EncoderDecoderModel.from_pretra
src/transformers/src/transformers/models/encoder_decoder/modeling_encoder_decoder.py:292
↓ 13 callersMethodfrom_pretrained
(cls, pretrained_model_name_or_path, **kwargs)
src/transformers/src/transformers/models/pop2piano/processing_pop2piano.py:137
↓ 13 callersMethodgenerate_speech
r""" Converts a sequence of input tokens into a sequence of mel spectrograms, which are subsequently turned into a speech waveform usi
src/transformers/src/transformers/models/speecht5/modeling_speecht5.py:2852
↓ 13 callersMethodget_context
(self)
src/transformers/src/transformers/models/sew_d/modeling_sew_d.py:644
↓ 13 callersFunctionget_device_map
Returns a dictionary of layers distributed evenly across all devices.
src/transformers/src/transformers/utils/model_parallel_utils.py:50
↓ 13 callersMethodget_encoder_decoder_model
(self, config, decoder_config)
src/transformers/tests/models/encoder_decoder/test_modeling_encoder_decoder.py:57
↓ 13 callersFunctionget_file_from_repo
Tries to locate a file in a local folder and repo, downloads and cache it if necessary. Args: path_or_repo (`str` or `os.PathLike`):
src/transformers/src/transformers/utils/hub.py:477
↓ 13 callersFunctionget_linear_schedule_with_warmup
Create a schedule with a learning rate that decreases linearly from the initial lr set in the optimizer to 0, after a warmup period during wh
src/transformers/src/transformers/optimization.py:108
↓ 13 callersMethodget_memory_footprint
r""" Get the memory footprint of a model. This will return the memory footprint of the current model in bytes. Useful to benchmark the
src/transformers/src/transformers/modeling_utils.py:2825
↓ 13 callersMethodget_nodes_and_xpaths_batch
(self)
src/transformers/tests/models/markuplm/test_tokenization_markuplm.py:78
↓ 13 callersFunctionget_resize_output_image_size
( input_image: np.ndarray, max_size: int = 448, input_data_format: Optional[Union[str, ChannelDime
src/transformers/src/transformers/models/tvp/image_processing_tvp.py:65
↓ 13 callersMethodget_special_tokens_mask
Retrieve sequence ids from a token list that has no special tokens added. This method is called when adding special tokens using the
src/transformers/src/transformers/models/t5/tokenization_t5.py:250
↓ 13 callersMethodget_tokenizer
(self, **kwargs)
src/transformers/tests/models/align/test_processor_align.py:77
↓ 13 callersMethodget_words_and_boxes_batch
(self)
src/transformers/tests/models/layoutlmv3/test_tokenization_layoutlmv3.py:67
↓ 13 callersMethodhf_compute_loss
CrossEntropyLoss that ignores pad tokens
src/transformers/src/transformers/models/led/modeling_tf_led.py:2636
← previousnext →701–800 of 49,163, ranked by callers