Functions — github.com/ModalityDance/Omni-R1

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

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

⨍ Functions 49,163◇ Types & classes 12,513↳ Endpoints 884

↓ 19 callersMethodfor_model

(cls, model_type: str, *args, **kwargs)

src/transformers/src/transformers/models/auto/configuration_auto.py:874

↓ 19 callersMethodfrom_pretrained

r""" Example: ```python >>> from transformers import MusicgenForConditionalGeneration >>> model = MusicgenForConditi

src/transformers/src/transformers/models/musicgen/modeling_musicgen.py:1806

↓ 19 callersMethodgenerate

r""" Generates sequences of token ids for models with a language modeling head. <Tip warning={true}> Most generation-contro

src/transformers/src/transformers/generation/utils.py:1587

↓ 19 callersMethodget_component

(self, attribute, **kwargs)

src/transformers/tests/test_processing_common.py:58

↓ 19 callersMethodget_eval_dataloader

Returns the evaluation [`~torch.utils.data.DataLoader`]. Subclass and override this method if you want to inject some custom behavio

src/transformers/src/transformers/trainer.py:943

↓ 19 callersMethodget_tokenizer

(self, **kwargs)

src/transformers/tests/models/flava/test_processor_flava.py:82

↓ 19 callersFunctionis_tensorboard_available

()

src/transformers/src/transformers/integrations/integration_utils.py:145

↓ 19 callersFunctionis_torch_fp16_available_on_device

(device)

src/transformers/src/transformers/utils/import_utils.py:478

↓ 19 callersFunctionis_torch_xpu_available

Checks if XPU acceleration is available either via `intel_extension_for_pytorch` or via stock PyTorch (>=2.4) and potentially if a XPU is in

src/transformers/src/transformers/utils/import_utils.py:815

↓ 19 callersMethodpad

Pad an image to make the height and width divisible by `size`. Args: image (`np.ndarray`): Image to pad.

src/transformers/src/transformers/models/swin2sr/image_processing_swin2sr.py:69

↓ 19 callersMethodprepare_config_and_inputs

(self, position_embeddings_type="relative")

src/transformers/tests/models/wav2vec2_bert/test_modeling_wav2vec2_bert.py:130

↓ 19 callersMethodprepare_config_and_inputs

(self)

src/transformers/tests/models/switch_transformers/test_modeling_switch_transformers.py:113

↓ 19 callersMethodprepare_config_and_inputs_for_common

(self)

src/transformers/tests/models/blip/test_modeling_blip.py:419

↓ 19 callersMethodreduce

Compute output projection inside each attention experts and merge the outputs of different experts.

src/transformers/src/transformers/models/jetmoe/modeling_jetmoe.py:386

↓ 19 callersFunctionreplace_default_in_arg_description

Catches the default value in the description of an argument inside a docstring and replaces it by the value passed. Args: descriptio

src/transformers/utils/check_docstrings.py:671

↓ 19 callersFunctionrequire_version_core

require_version wrapper which emits a core-specific hint on failure

src/transformers/src/transformers/utils/versions.py:114

↓ 19 callersMethodto_dict

Serializes this instance to a Python dictionary. Override the default [`~PretrainedConfig.to_dict`]. Returns: `Dict[str,

src/transformers/src/transformers/models/bark/generation_configuration_bark.py:317

↓ 18 callersMethod__init__

(self, config)

src/transformers/src/transformers/models/altclip/modeling_altclip.py:827

↓ 18 callersMethod__init__

(self, config: RobertaConfig, **kwargs)

src/transformers/src/transformers/models/roberta/modeling_tf_roberta.py:482

↓ 18 callersMethod__init__

(self, config: HubertConfig)

src/transformers/src/transformers/models/hubert/modeling_hubert.py:1232

↓ 18 callersMethod__init__

(self, config: RobertaPreLayerNormConfig, **kwargs)

src/transformers/src/transformers/models/roberta_prelayernorm/modeling_tf_roberta_prelayernorm.py:483

↓ 18 callersMethod__init__

(self, config: RemBertConfig, **kwargs)

src/transformers/src/transformers/models/rembert/modeling_tf_rembert.py:426

↓ 18 callersMethod__init__

(self, config)

src/transformers/src/transformers/models/rembert/modeling_rembert.py:411

↓ 18 callersMethod__init__

(self, config)

src/transformers/src/transformers/models/xmod/modeling_xmod.py:426

↓ 18 callersMethod__init__

(self, config)

src/transformers/src/transformers/models/deberta/modeling_deberta.py:358

↓ 18 callersMethod__init__

(self, config, layer_id=0)

src/transformers/src/transformers/models/reformer/modeling_reformer.py:1410

↓ 18 callersMethod__init__

(self, config: CamembertConfig, **kwargs)

src/transformers/src/transformers/models/camembert/modeling_tf_camembert.py:581

↓ 18 callersMethod__init__

(self, matcher, num_classes, eos_coef, losses)

src/transformers/src/transformers/models/yolos/modeling_yolos.py:972

↓ 18 callersMethod__init__

(self, config)

src/transformers/src/transformers/models/vilt/modeling_vilt.py:469

↓ 18 callersMethod__init__

(self, config: XLMRobertaConfig, **kwargs)

src/transformers/src/transformers/models/xlm_roberta/modeling_tf_xlm_roberta.py:572

↓ 18 callersMethod__init__

(self, config: MobileViTConfig, **kwargs)

src/transformers/src/transformers/models/mobilevit/modeling_tf_mobilevit.py:1131

↓ 18 callersMethod__init__

(self, config, has_relative_attention_bias=False)

src/transformers/src/transformers/models/udop/modeling_udop.py:906

↓ 18 callersMethod__init__

(self, config)

src/transformers/src/transformers/models/visual_bert/modeling_visual_bert.py:343

↓ 18 callersMethod__init__

(self, config)

src/transformers/src/transformers/models/chameleon/modeling_chameleon.py:254

↓ 18 callersMethod__init__

(self, config)

src/transformers/src/transformers/models/convbert/modeling_convbert.py:535

↓ 18 callersMethod__init__

(self, config)

src/transformers/src/transformers/models/electra/modeling_electra.py:445

↓ 18 callersMethod__init__

(self, config, block_index, **kwargs)

src/transformers/src/transformers/models/funnel/modeling_tf_funnel.py:597

↓ 18 callersMethod__init__

( self, config, dim, input_resolution: Tuple[int, int], num_heads: int, shift_size: int = 0, **kwargs

src/transformers/src/transformers/models/swin/modeling_tf_swin.py:744

↓ 18 callersMethod__init__

(self, **kwargs)

src/transformers/src/transformers/models/deprecated/efficientformer/modeling_tf_efficientformer.py:427

↓ 18 callersMethod__init__

( self, config, local, always_attend_to_first_position, first_position

src/transformers/src/transformers/models/canine/modeling_canine.py:685

↓ 18 callersMethod_prepare_for_class

(self, inputs_dict, model_class, return_labels=False)

src/transformers/tests/models/grounding_dino/test_modeling_grounding_dino.py:221

↓ 18 callersMethodbuild

(self, input_shape: tf.TensorShape = None)

src/transformers/src/transformers/models/convnextv2/modeling_tf_convnextv2.py:92

↓ 18 callersMethodcheck_results_dict_not_empty

(self, results)

src/transformers/tests/benchmark/test_benchmark_tf.py:32

↓ 18 callersFunctionclassifier

(text, labels)

src/transformers/src/transformers/agents/evaluate_agent.py:22

↓ 18 callersMethoddtype

Returns the dtype of the underlying rotation. Returns: The dtype of the underlying rotation

src/transformers/src/transformers/models/esm/openfold_utils/rigid_utils.py:412

↓ 18 callersFunctionexecute_subprocess_async

(cmd, env=None, stdin=None, timeout=180, quiet=False, echo=True)

src/transformers/src/transformers/testing_utils.py:2042

↓ 18 callersMethodforward

( self, input_ids: torch.LongTensor = None, )

src/transformers/tests/models/phi3/test_modeling_phi3.py:61

↓ 18 callersMethodget_config_dict

(self, stage)

src/transformers/tests/deepspeed/test_deepspeed.py:415

↓ 18 callersMethodget_env

Return a copy of the `os.environ` object that sets up `PYTHONPATH` correctly, depending on the test suite it's invoked from. This is

src/transformers/src/transformers/testing_utils.py:1642

↓ 18 callersMethodget_from_offsets

(offsets, key)

src/transformers/tests/models/wav2vec2/test_tokenization_wav2vec2.py:494

↓ 18 callersFunctionids_tensor

(shape, vocab_size, rng=None, name=None)

src/transformers/tests/test_modeling_common.py:4830

↓ 18 callersMethodprepare_config_and_inputs

(self)

src/transformers/tests/models/bert/test_modeling_bert.py:103

↓ 18 callersFunctionspectrogram_batch

Calculates spectrograms for a list of waveforms using the Short-Time Fourier Transform, optimized for batch processing. This function extends

src/transformers/src/transformers/audio_utils.py:584

↓ 18 callersMethodword_to_tokens

Get the encoded token span corresponding to a word in a sequence of the batch. Token spans are returned as a [`~tokenization_utils_b

src/transformers/src/transformers/tokenization_utils_base.py:467

↓ 17 callersMethod__class__

(self)

src/transformers/src/transformers/utils/fx.py:722

↓ 17 callersMethod__init__

(self, config)

src/transformers/src/transformers/models/yoso/modeling_yoso.py:516

↓ 17 callersMethod__init__

(self, config)

src/transformers/src/transformers/models/xlm_roberta_xl/modeling_xlm_roberta_xl.py:369

↓ 17 callersMethod__init__

(self, config)

src/transformers/src/transformers/models/roberta/modeling_roberta.py:384

↓ 17 callersMethod__init__

(self, config, name=None)

src/transformers/src/transformers/models/esm/modeling_tf_esm.py:574

↓ 17 callersMethod__init__

(self, config)

src/transformers/src/transformers/models/roberta_prelayernorm/modeling_roberta_prelayernorm.py:377

↓ 17 callersMethod__init__

(self, config, layer_idx=None)

src/transformers/src/transformers/models/git/modeling_git.py:346

↓ 17 callersMethod__init__

(self, config, layer_id=0, **kwargs)

src/transformers/src/transformers/models/longformer/modeling_tf_longformer.py:1607

↓ 17 callersMethod__init__

(self, config)

src/transformers/src/transformers/models/data2vec/modeling_data2vec_text.py:388

↓ 17 callersMethod__init__

(self, config)

src/transformers/src/transformers/models/mra/modeling_mra.py:704

↓ 17 callersMethod__init__

(self, config: ChineseCLIPConfig)

src/transformers/src/transformers/models/chinese_clip/modeling_chinese_clip.py:1340

↓ 17 callersMethod__init__

Creates an index Args: indices (`torch.LongTensor`, same shape as a *values* Tensor to which the indices refer):

src/transformers/src/transformers/models/tapas/modeling_tapas.py:1556

↓ 17 callersMethod__init__

(self, config)

src/transformers/src/transformers/models/camembert/modeling_camembert.py:402

↓ 17 callersMethod__init__

(self, config)

src/transformers/src/transformers/models/nystromformer/modeling_nystromformer.py:315

↓ 17 callersMethod__init__

(self)

src/transformers/src/transformers/models/squeezebert/modeling_squeezebert.py:95

↓ 17 callersMethod__init__

(self, config)

src/transformers/src/transformers/models/xlm_roberta/modeling_xlm_roberta.py:385

↓ 17 callersMethod__init__

(self, config: MobileViTConfig)

src/transformers/src/transformers/models/mobilevit/modeling_mobilevit.py:896

↓ 17 callersMethod__init__

(self, config)

src/transformers/src/transformers/models/x_clip/modeling_x_clip.py:277

↓ 17 callersMethod__init__

(self, config)

src/transformers/src/transformers/models/deprecated/tvlt/modeling_tvlt.py:489

↓ 17 callersMethod__init__

(self)

src/transformers/src/transformers/models/deprecated/efficientformer/modeling_efficientformer.py:277

↓ 17 callersMethod__init__

(self, config: LayoutLMConfig, **kwargs)

src/transformers/src/transformers/models/layoutlm/modeling_tf_layoutlm.py:470

↓ 17 callersMethod__init__

(self, matcher, num_classes, eos_coef, losses)

src/transformers/src/transformers/models/detr/modeling_detr.py:1942

↓ 17 callersMethod_get_uniform_logits

(self, batch_size: int, length: int)

src/transformers/tests/generation/test_tf_logits_process.py:52

↓ 17 callersMethod_init_backbone

Method to initialize the backbone. This method is called by the constructor of the base class after the pretrained model weights have

src/transformers/src/transformers/utils/backbone_utils.py:179

↓ 17 callersMethodapply_gradients

(self, grads_and_vars, name=None, **kwargs)

src/transformers/src/transformers/optimization_tf.py:258

↓ 17 callersMethodbuild

(self, input_shape=None)

src/transformers/src/transformers/models/gptj/modeling_tf_gptj.py:308

↓ 17 callersMethodbuild

(self, input_shape=None)

src/transformers/src/transformers/models/opt/modeling_tf_opt.py:914

↓ 17 callersFunctioncheck_json_file_has_correct_format

(file_path)

src/transformers/src/transformers/testing_utils.py:2117

↓ 17 callersMethodconvert

(k, v)

src/transformers/examples/research_projects/seq2seq-distillation/_test_seq2seq_examples_multi_gpu.py:130

↓ 17 callersFunctionconvert_pytorch_state_dict_to_flax

(pt_state_dict, flax_model)

src/transformers/src/transformers/modeling_flax_pytorch_utils.py:154

↓ 17 callersMethodfill_with_past_key_values_

(self, inputs_or_outputs: Mapping[str, Mapping[int, str]], direction: str)

src/transformers/src/transformers/models/blenderbot/configuration_blenderbot.py:378

↓ 17 callersMethodfrom_pretrained

(cls, pretrained_model_name_or_path: Union[str, os.PathLike], **kwargs)

src/transformers/src/transformers/models/siglip/configuration_siglip.py:114

↓ 17 callersMethodget_lr

(self)

src/transformers/src/transformers/optimization.py:931

↓ 17 callersFunctionget_results

(output_dir)

src/transformers/examples/pytorch/test_pytorch_examples.py:86

↓ 17 callersFunctionis_torch_npu_available

Checks if `torch_npu` is installed and potentially if a NPU is in the environment

src/transformers/src/transformers/utils/import_utils.py:640

↓ 17 callersMethodnormalize

( self, input_features: List[np.ndarray], attention_mask: Optional[np.ndarray] = None )

src/transformers/src/transformers/models/speech_to_text/feature_extraction_speech_to_text.py:153

↓ 17 callersFunctionpermute_final_dims

(tensor: torch.Tensor, inds: List[int])

src/transformers/src/transformers/models/esm/modeling_esmfold.py:221

↓ 17 callersMethodprepare_config_and_inputs

(self)

src/transformers/tests/models/bert/test_modeling_tf_bert.py:97

↓ 17 callersMethodquestion_encoder

(self)

src/transformers/src/transformers/models/rag/modeling_rag.py:1200

↓ 17 callersFunctionrescale

Rescales `image` by `scale`. Args: image (`np.ndarray`): The image to rescale. scale (`float`): The

src/transformers/src/transformers/image_transforms.py:97

↓ 17 callersMethodsave_pretrained

(self, save_directory)

src/transformers/src/transformers/models/wav2vec2_with_lm/processing_wav2vec2_with_lm.py:115

↓ 17 callersMethodset_defaults

(cls, prefix, defaults)

src/transformers/src/transformers/utils/hp_naming.py:25

↓ 17 callersMethodto_dict

Serializes this instance to a Python dictionary. Returns: `Dict[str, Any]`: Dictionary of all the attributes that make u

src/transformers/src/transformers/models/clap/feature_extraction_clap.py:137

↓ 17 callersFunctionto_pil_image

Converts `image` to a PIL Image. Optionally rescales it and puts the channel dimension back as the last axis if needed. Args: im

src/transformers/src/transformers/image_transforms.py:162

↓ 17 callersMethodtokenizer

(self)

src/transformers/tests/models/pegasus/test_modeling_tf_pegasus.py:225

↓ 17 callersMethodupdate

Reads in a token and returns booleans that indicate the progress made by it. This function will update the state of this object unlik

src/transformers/src/transformers/generation/beam_constraints.py:70

← previous next →501–600 of 49,163, ranked by callers