Utility method to check if the model has correctly adapters injected on it.

Creates a causal 4D mask of shape `(batch_size, 1, query_length, key_value_length)` from a 2D mask of shape `(batch_size, key_value_length)`

splits sequence length dim of vectors into `dim_factor_1` and `dim_factor_2` dims

Builds a event of a streaming OpenAI Response response. IMPORTANT: The serialized chunk won't contain empty fields (fields with `Non

Convert the inner content to tensors. Args: tensor_type (`str` or [`~utils.TensorType`], *optional*): Th

query_states: text key_value_states: protein query_states: [bs, query_seq_len, dim] key_value_states: [bs, kv_seq_len

A utility method that massages the config file and can optionally verify that the values match. 1. Replace "auto" values with `Train

Generates audio from an input prompt and an additional optional `Bark` speaker prompt. Args: input_ids (`Optional[torch.

Returns the generation mode triggered by the [`GenerationConfig`] instance. Arg: assistant_model (`PreTrainedModel`, *op

Returns the optimizer class and optimizer parameters based on the training arguments. Args: args (`transformers.training

Same as `torch.isin` without flags, but MPS-friendly. We can remove this function when we stop supporting torch <= 2.3. See https://github.co

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

Converts the raw output of [`Owlv2ForObjectDetection`] into final bounding boxes in (top_left_x, top_left_y, bottom_right_x, bottom_r

Start monitoring GPU metrics.

Computes random mask spans for a given shape. Used to implement [SpecAugment: A Simple Data Augmentation Method for ASR](https://huggingface.

Patch a callerable `attr_name` of a module or class `module_or_class`. This will allow us to collect the call information, e.g. the argument

Returns a mask function that is the intersection of provided mask functions

Similar to the `apply_chat_template` method on tokenizers, this method applies a Jinja template to input conversations to turn them i

Builds a chunk of a streaming OpenAI Chat Completion response. IMPORTANT: The serialized chunk won't contain empty fields (fields wi

Calculate rouge using rouge_scorer package. Args: pred_lns: list of summaries generated by model tgt_lns: list of groundtruth sum

Delete a PEFT adapter from the underlying model. Args: adapter_names (`Union[list[str], str]`): The name

r""" Converts a sequence of input tokens into a sequence of mel spectrograms, which are subsequently turned into a speech waveform usi

Given an image and a target resolution, calculate the output size of the image after cropping to the target

Find the target (height, width) dimension of the output image after resizing given the input image and the desired size. Args: i